added the missing positional parameter for _create_cehrbert_data_from…

…_meds

src/cehrbert/data_generators/hf_data_generator/meds_utils.py

@@ -34,7 +34,7 @@ def get_meds_to_cehrbert_conversion_cls(
     meds_to_cehrbert_conversion_type: MedsToCehrBertConversionType,
 ) -> MedsToCehrBertConversion:
     for cls in MedsToCehrBertConversion.__subclasses__():
-        if meds_to_cehrbert_conversion_type.name == cls.__name__:
+        if meds_to_cehrbert_conversion_type.__name__ == cls.__name__:
             return cls()
     raise RuntimeError(f"{meds_to_cehrbert_conversion_type} is not a valid MedsToCehrBertConversionType")
 
@@ -445,6 +445,7 @@ def _create_cehrbert_data_from_meds(
         _meds_to_cehrbert_generator,
         path_to_db=data_args.data_folder,
         default_visit_id=default_visit_id,
+        meds_to_cehrbert_conversion_type=data_args.meds_to_cehrbert_conversion_type,
     )
     dataset = Dataset.from_generator(
         batch_func,