Merge pull request #12 from cms-ml/fix/tf_tools_import

Fix wrong import in aot compat script.
cms-ml · Dec 11, 2023 · 6214bbf · 6214bbf
2 parents 5666bba + 0a64f77
commit 6214bbf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmsml/scripts/check_aot_compatibility.py b/cmsml/scripts/check_aot_compatibility.py
@@ -10,7 +10,8 @@
 import tabulate
 
 from cmsml.util import colored
-from cmsml.tensorflow.aot import OpsData, load_graph_def, get_graph_ops
+from cmsml.tensorflow.tools import load_graph_def
+from cmsml.tensorflow.aot import OpsData, get_graph_ops
 
 
 def check_aot_compatibility(