fix data ndarray not list

deepmodeling · Jun 24, 2024 · 842a420 · 842a420
1 parent 4feab87
commit 842a420
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unimol_tools/unimol_tools/data/datareader.py b/unimol_tools/unimol_tools/data/datareader.py
@@ -59,7 +59,7 @@ def read_data(self, data=None, is_train=True, **params):
             _ = data.pop('target', None)
             data = pd.DataFrame(data).rename(columns={smiles_col: 'SMILES'})
 
-        elif isinstance(data, list):
+        elif isinstance(data, list) or isinstance(data, np.ndarray):
             # load from smiles list
             data = pd.DataFrame(data, columns=['SMILES'])
         else: