Merge pull request #151 from raimis/fix_2

Fix a missing argument for isinstance
torchmd · Nov 15, 2022 · df7c906 · df7c906
2 parents 9e0fef3 + 7192f9d
commit df7c906
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/models/model.py b/torchmdnet/models/model.py
@@ -127,7 +127,7 @@ def create_prior_models(args, dataset=None):
                 prior_args.append({})
         if 'prior_args' in args:
             prior_args = args['prior_args']
-            if not isinstance(prior_args):
+            if not isinstance(prior_args, list):
                 prior_args = [prior_args]
         for name, arg in zip(prior_names, prior_args):
             assert hasattr(priors, name), (