code: def test_0_embedding_at_inference_time(self):
      df = pd.read_csv("tests/data/hdi.csv")
      pdef = ProblemDefinition.from_dict(
          {"target": "Development Index", "time_aim": 10}
      )
      predictor = predictor_from_problem(df, pdef)
      train_and_check_time_aim(predictor, df, ignore_time_aim=True)
      predictions = predictor.predict(
          df, args={"return_embedding": True}
      )
  
      self.assertTrue(predictions.shape[0] == len(df))
      self.assertTrue(predictions.shape[1] != 1)
  
file_path: lightwood/tests/integration/basic/test_embedding.py
imports: import unittest
  import pandas
  from tests.utils.timing import train_and_check_time_aim
  from lightwood.api.types import ProblemDefinition
  from lightwood.api.high_level import predictor_from_problem
label_desc: Use image node_id 1 for calling the TestEmbeddingPredictor obj's underlying member method code with example usage: obj.test_0_embedding_at_inference_time() without return types