fixed the missing required arguments

Author: ChaoPang

src/cehrbert/data_generators/hf_data_generator/hf_dataset_mapping.py

@@ -345,13 +345,10 @@ def transform(self, record: Dict[str, Any]) -> Dict[str, Any]:
             year_str = f"year:{str(first_visit_start_datetime.year)}"
             age_str = f"age:{str(relativedelta(first_visit_start_datetime, birth_datetime).years)}"
 
-            self._update_cehrbert_record(
-                cehrbert_record,
-                year_str,
-            )
-            self._update_cehrbert_record(cehrbert_record, age_str)
-            self._update_cehrbert_record(cehrbert_record, gender)
-            self._update_cehrbert_record(cehrbert_record, race)
+            self._update_cehrbert_record(cehrbert_record, year_str, first_visit_start_datetime)
+            self._update_cehrbert_record(cehrbert_record, age_str, first_visit_start_datetime)
+            self._update_cehrbert_record(cehrbert_record, gender, first_visit_start_datetime)
+            self._update_cehrbert_record(cehrbert_record, race, first_visit_start_datetime)
 
         # A bool indicator to toggle between 1 and 2
         visit_segment_indicator = False

tests/unit_tests/data_generators/hf_data_generator/hf_med_to_cehrbert_mapping_test.py

@@ -36,6 +36,7 @@ def setUp(self):
             birth_datetime=datetime(1980, 4, 14, 0, 0),
             gender="Gender/F",
             race="Race/unknown",
+            ethnicity="Ethnicity/unknown",
             visits=[inpatient_visit, outpatient_visit],
         )