fix: missing arg in yaml_file_has_column_of_type 3

OpenBioML · Nov 20, 2023 · 3befcd0 · 3befcd0
1 parent 762ceae
commit 3befcd0
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 177 deletions.
diff --git a/data/kg/compound_protein_compound/meta.yaml b/data/kg/compound_protein_compound/meta.yaml
diff --git a/data/kg/compound_protein_go_term/meta.yaml b/data/kg/compound_protein_go_term/meta.yaml
diff --git a/data/tabular/train_test_split.py b/data/tabular/train_test_split.py
@@ -467,7 +467,7 @@ def assign_splits(
         file
         for file in all_yaml_files
         if (
-            yaml_file_has_column_of_type(file)
+            yaml_file_has_column_of_type(file, "SMILES")
             and not yaml_file_has_column_of_type(file, "AS_SEQUENCE")
         )  # noqa: E501
     ]