Merge pull request deepchem#3718 from GreatRSingh/patch-2

error fix in safeops
GreatRSingh · Dec 14, 2023 · ce19366 · ce19366
2 parents 7736726 + 0fabbe5
commit ce19366
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepchem/utils/safeops_utils.py b/deepchem/utils/safeops_utils.py
@@ -1,9 +1,9 @@
 import math
 import torch
 from typing import Union, Optional, Tuple
-from deepchem.utils.dft_utils import ZType
 
 eps = 1e-12
+ZType = Union[int, float, torch.Tensor]
 
 # safe operations