:fix: pytorch cannot convert to onnx

cap-ntu · Apr 17, 2021 · 38b7728 · 38b7728
1 parent ce38060
commit 38b7728
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modelci/types/type_conversion.py b/modelci/types/type_conversion.py
@@ -102,7 +102,7 @@ def model_data_type_to_np(model_dtype):
 
 
 def model_data_type_to_torch(model_dtype):
-    from modelci.types.bo import DataType
+    from modelci.types.models.common import DataType
     import torch
 
     mapper = {