Add SnitchSim to Python packages

pulp-platform · Aug 10, 2024 · dd6243a · dd6243a
1 parent aa778a7
commit dd6243a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/target/common/test/SnitchSim.py → util/sim/SnitchSim.py b/target/common/test/SnitchSim.py → util/sim/SnitchSim.py
diff --git a/util/sim/__init__.py b/util/sim/__init__.py
@@ -0,0 +1,9 @@
+# Copyright 2024 ETH Zurich and University of Bologna.
+# Licensed under the Apache License, Version 2.0, see LICENSE for details.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Luca Colagrande <[email protected]>
+
+from .SnitchSim import SnitchSim
+
+__all__ = ['SnitchSim']
diff --git a/util/sim/verif_utils.py b/util/sim/verif_utils.py
@@ -7,16 +7,14 @@
 """Convenience functions and classes for verification scripts."""
 
 
-import sys
 import argparse
 import numpy as np
 import csv
-from snitch.util.sim.Elf import Elf
 from pathlib import Path
-from snitch.util.sim.data_utils import flatten, from_buffer
 
-sys.path.append(str(Path(__file__).parent / '../../target/common/test/'))
-from SnitchSim import SnitchSim  # noqa: E402
+from snitch.util.sim.Elf import Elf
+from snitch.util.sim.data_utils import flatten, from_buffer
+from snitch.util.sim import SnitchSim 
 
 
 def dump_results_to_csv(expected_results, actual_results, error, max_error, path):