Updated importing of ptype

ai2es · Sep 21, 2023 · 47af4ab · 47af4ab
1 parent 1b1c869
commit 47af4ab
Show file tree

Hide file tree

Showing 2 changed files with 523 additions and 93 deletions.
diff --git a/applications/train_classifier_ptype.py b/applications/train_classifier_ptype.py
@@ -15,10 +15,15 @@
 from tensorflow.keras import backend as K
 from argparse import ArgumentParser
 
-from ptype.callbacks import MetricsCallback
-from ptype.data import load_ptype_uq, preprocess_data
-from sklearn.model_selection import GroupShuffleSplit
+try:
+    from ptype.callbacks import MetricsCallback
+except ImportError:
+    import subprocess
+    subprocess.run(['pip', 'install', 'git+https://github.com/ai2es/ptype-physical.git'], check=True)
+    from ptype.callbacks import MetricsCallback
+    from ptype.data import load_ptype_uq, preprocess_data
 
+from sklearn.model_selection import GroupShuffleSplit
 from evml.keras.callbacks import get_callbacks, ReportEpoch
 from evml.keras.models import CategoricalDNN
 from evml.pbs import launch_pbs_jobs

diff --git a/notebooks/regression_example.ipynb b/notebooks/regression_example.ipynb