fix: outdated imports

elasticai/creator/base_modules/layers.py

@@ -1,10 +1,10 @@
 from typing import Callable, Optional, cast
 
 import torch
+from torch import Tensor
+from torch.nn import Module
 from torch.nn.utils.parametrize import register_parametrization
 
-from elasticai.creator.mlframework import Module, Tensor
-
 from .autograd_functions.binary_quantization import Binarize as BinarizeFn
 
 """Implementation of quantizers and quantized variants of pytorch layers"""

elasticai/creator/nn/conv1d/design.py

@@ -7,8 +7,8 @@
 )
 from elasticai.creator.vhdl.design.design import Design
 from elasticai.creator.vhdl.design.ports import Port
-from elasticai.creator.vhdl.designs.rom import Rom
 from elasticai.creator.vhdl.savable import Path
+from elasticai.creator.vhdl.shared_designs.rom import Rom
 
 
 class FPConv1d(Design):

elasticai/creator/nn/linear/design.py

@@ -8,8 +8,8 @@
 )
 from elasticai.creator.vhdl.design.design import Design
 from elasticai.creator.vhdl.design.ports import Port
-from elasticai.creator.vhdl.designs.rom import Rom
 from elasticai.creator.vhdl.savable import Path
+from elasticai.creator.vhdl.shared_designs.rom import Rom
 
 
 class FPLinear(Design):

elasticai/creator/nn/precomputed/fp_precomputed_module.py

@@ -12,7 +12,7 @@
     FixedPointConfig,
 )
 from elasticai.creator.vhdl.design.design import Design
-from elasticai.creator.vhdl.designs.precomputed_scalar_function import (
+from elasticai.creator.vhdl.shared_designs.precomputed_scalar_function import (
     PrecomputedScalarFunction,
 )
 from elasticai.creator.vhdl.translatable import Translatable

tests/vhdl/designs/rom/test_rom_file.py

@@ -5,7 +5,7 @@
 import pytest
 
 from elasticai.creator.in_memory_path import InMemoryFile, InMemoryPath
-from elasticai.creator.vhdl.designs.rom import Rom
+from elasticai.creator.vhdl.shared_designs.rom import Rom
 
 from .utils import extract_rom_values