error with scVI search compliance with pytorchlightining

Author: saroudant

sobolev_alignment/scvi_model_search.py

@@ -12,6 +12,7 @@
 from anndata import AnnData
 from hyperopt import STATUS_FAIL, STATUS_OK, Trials, fmin, hp, tpe
 from sklearn.model_selection import train_test_split
+from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
 from ._scvi_default_params import SCVI_MODEL_PARAMS, SCVI_PLAN_PARAMS, SCVI_TRAIN_PARAMS
 
@@ -179,6 +180,9 @@ def _objective_function(params):
         except ValueError:
             results_dict = {"status": STATUS_FAIL, "loss": np.iinfo(np.uint64).max}
             results_dict.update(params)
+        except MisconfigurationException:
+            results_dict = {"status": STATUS_FAIL, "loss": np.iinfo(np.uint64).max}
+            results_dict.update(params)
 
         results_dict.update(params)
         return results_dict