Fix slothy imports in ntt helium script

slothy-optimizer · Dec 23, 2023 · 40af412 · 40af412
1 parent 95b52f9
commit 40af412
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/paper/scripts/slothy_ntt_helium.py b/paper/scripts/slothy_ntt_helium.py
@@ -28,12 +28,11 @@
 import argparse, logging, sys, os, time
 from io import StringIO
 
-from slothy.slothy import Slothy
-from slothy.core import Config
+from slothy import Slothy, Config
 
-import targets.arm_v81m.arch_v81m as Arch_Armv81M
-import targets.arm_v81m.cortex_m55r1 as Target_CortexM55r1
-import targets.arm_v81m.cortex_m85r1 as Target_CortexM85r1
+import slothy.targets.arm_v81m.arch_v81m as Arch_Armv81M
+import slothy.targets.arm_v81m.cortex_m55r1 as Target_CortexM55r1
+import slothy.targets.arm_v81m.cortex_m85r1 as Target_CortexM85r1
 
 target_label_dict = {Target_CortexM55r1: "m55",
                      Target_CortexM85r1: "m85"}