Merge pull request deepchem#3632 from MarcS18/master

Atomic Contributions bug fix
GreatRSingh · Nov 1, 2023 · 56adfb8 · 56adfb8
2 parents 8910cda + 91235dd
commit 56adfb8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/deepchem/data/data_loader.py b/deepchem/data/data_loader.py
@@ -908,7 +908,11 @@ def _featurize_shard(self,
         features = [
             elt for (is_valid, elt) in zip(valid_inds, features) if is_valid
         ]
-        return np.array(features), valid_inds
+        try:
+            return np.array(features), valid_inds
+        except ValueError as e:
+            logger.warning("Exception message: {}".format(e))
+            return np.asarray(features, dtype=object), valid_inds
 
 
 class FASTALoader(DataLoader):