delete assert

model.py

@@ -68,7 +68,6 @@ def main():
     testing_data = pd.read_csv(flags.input)
 
     # extract samples and labels
-    # assert 'aki' in training_data.columns, "Error: Training data must contain trained labels."
     training_x = training_data.drop(columns=['aki'])
     # Ignore 'aki' if not present in test data
     testing_x = testing_data.drop(columns=['aki'], errors='ignore')