Fix some imports

Author: TimKoornstra

tests/test_parser/test_tf_to_spec.py

@@ -1,7 +1,7 @@
 import pytest
 import tensorflow as tf
 
-from vgslify.parsers.tensorflow import tf_to_spec
+from vgslify.model_parsers.tensorflow import tf_to_spec
 
 
 def test_input_layer():

vgslify/model_parsers/tensorflow.py

@@ -18,7 +18,7 @@
     ReshapeConfig,
     RNNConfig,
 )
-from vgslify.parsers.base import BaseModelParser
+from vgslify.model_parsers.base import BaseModelParser
 
 
 class TensorFlowModelParser(BaseModelParser):
@@ -426,7 +426,7 @@ def register_custom_parser(layer_cls: Type[tf.keras.layers.Layer]) -> Callable:
     --------
     Registering a custom parser for a `MyCustomLayer`:
 
-    >>> from vgslify.parsers.tensorflow import register_custom_parser
+    >>> from vgslify.model_parsers.tensorflow import register_custom_parser
     >>> import tensorflow as tf
     >>> class MyCustomLayer(tf.keras.layers.Layer):
     ...     def __init__(self, units: int):

vgslify/model_parsers/torch.py

@@ -19,7 +19,7 @@
     ReshapeConfig,
     RNNConfig,
 )
-from vgslify.parsers.base import BaseModelParser
+from vgslify.model_parsers.base import BaseModelParser
 from vgslify.torch.layers import Reshape
 
 
@@ -431,7 +431,7 @@ def register_custom_parser(layer_cls: Type[nn.Module]):
     --------
     Registering a custom parser for a `MyCustomLayer`:
 
-    >>> from vgslify.parsers.torch import register_custom_parser
+    >>> from vgslify.model_parsers.torch import register_custom_parser
     >>> from torch import nn
     >>> class MyCustomLayer(nn.Module):
     ...     def __init__(self, param: int):

vgslify/utils/model_to_spec.py

@@ -30,7 +30,7 @@ def model_to_spec(model) -> str:
 
     Examples
     --------
-    >>> from vgslify.utils import model_to_spec
+    >>> from vgslify import model_to_spec
     >>> import tensorflow as tf
     >>> model = tf.keras.models.load_model("path_to_model.h5")
     >>> spec_string = model_to_spec(model)
@@ -39,13 +39,13 @@ def model_to_spec(model) -> str:
 
     # Check if it's a TensorFlow model
     if tf and isinstance(model, tf.keras.Model):
-        from vgslify.parsers.tensorflow import TensorFlowModelParser
+        from vgslify.model_parsers import TensorFlowModelParser
 
         parser = TensorFlowModelParser()
 
     # Check if it's a PyTorch model
     if nn and isinstance(model, nn.Module):
-        from vgslify.parsers.torch import TorchModelParser
+        from vgslify.model_parsers import TorchModelParser
 
         parser = TorchModelParser()