fix: variable to catch return from preparte data (no of data points)

OpenBioML · Oct 8, 2023 · 78f7c5c · 78f7c5c
1 parent 75f0fe9
commit 78f7c5c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/data/tabular/qm9/transform.py b/data/tabular/qm9/transform.py
@@ -48,5 +48,7 @@ def prepare_data():
 
 if __name__ == "__main__":
     print(f" Preparing  clean tabular {DATASET_NAME} datatset")
-    prepare_data()
-    print(f" Finished Preparing  clean tabular {DATASET_NAME} datatset")
+    datapoints = prepare_data()
+    print(
+        f" Finished Preparing  clean tabular {DATASET_NAME} datatset with {datapoints} datapoints"
+    )