Update

Matheus-Monteiro · Dec 26, 2020 · 1bfb1cc · 1bfb1cc
1 parent 05ae3a7
commit 1bfb1cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepchem/models/torch_models/torch_model.py b/deepchem/models/torch_models/torch_model.py
@@ -364,7 +364,7 @@ def fit_generator(self,
 
       # Execute the loss function, accumulating the gradients.
 
-      if len(inputs) == 1:
+      if isinstance(inputs, list) and len(inputs) == 1:
         inputs = inputs[0]
 
       optimizer.zero_grad()
@@ -524,7 +524,7 @@ def _predict(
       inputs, _, _ = self._prepare_batch((inputs, None, None))
 
       # Invoke the model.
-      if len(inputs) == 1:
+      if isinstance(inputs, list) and len(inputs) == 1:
         inputs = inputs[0]
       output_values = self.model(inputs)
       if isinstance(output_values, torch.Tensor):