Skip to content

Commit 6214bbf

Browse files
authored
Merge pull request #12 from cms-ml/fix/tf_tools_import
Fix wrong import in aot compat script.
2 parents 5666bba + 0a64f77 commit 6214bbf

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

cmsml/scripts/check_aot_compatibility.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,8 @@
1010
import tabulate
1111

1212
from cmsml.util import colored
13-
from cmsml.tensorflow.aot import OpsData, load_graph_def, get_graph_ops
13+
from cmsml.tensorflow.tools import load_graph_def
14+
from cmsml.tensorflow.aot import OpsData, get_graph_ops
1415

1516

1617
def check_aot_compatibility(

0 commit comments

Comments
 (0)