netloader 3.11.0__py3-none-any.whl

netloader/__init__.py

@@ -0,0 +1,61 @@
+"""
+Package information, creates the logger and adds netloader classes to PyTorch safe globals
+"""
+import sys
+import logging
+import warnings
+
+
+__version__ = '3.11.0'
+__author__ = 'Ethan Tregidga'
+logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.WARNING)
+warnings.filterwarnings('once', category=DeprecationWarning, module=r'^netloader(\.|$)')
+warnings.filterwarnings(
+    'once',
+    category=PendingDeprecationWarning,
+    module=r'^netloader(\.|$)',
+)
+
+
+if sys.version_info < (3, 12):
+    warnings.warn(
+        f'Python version ({sys.version_info.major}.{sys.version_info.minor}) is '
+        f'deprecated, please upgrade to Python 3.12 or higher',
+        DeprecationWarning,
+    )
+
+try:
+    import torch
+
+    from netloader.utils import safe_globals
+    from netloader import (
+        layers,
+        models,
+        architectures,
+        utils,
+        data,
+        loss_funcs,
+        network,
+        schedulers,
+        transforms,
+    )
+
+
+    # Adds PyTorch Network Loader classes to list of safe PyTorch classes when loading saved
+    # architectures
+    safe_globals(__name__, [models, architectures, loss_funcs, schedulers, transforms])
+    torch.serialization.add_safe_globals([network.Network, network.CompatibleNetwork])
+
+    __all__ = [
+        'utils',
+        'layers',
+        'models',
+        'architectures',
+        'data',
+        'network',
+        'loss_funcs',
+        'schedulers',
+        'transforms',
+    ]
+except (ModuleNotFoundError, ImportError):
+    pass

netloader/architectures/__init__.py

@@ -0,0 +1,48 @@
+"""
+Collects all architectures
+"""
+from typing import Any
+from warnings import warn
+
+from netloader.architectures.utils import UtilityMixin
+from netloader.architectures.base import BaseArchitecture, load_arch
+from netloader.architectures.encoder_decoder import Autoencoder, Decoder, Encoder
+
+
+__all__ = [
+    'BaseArchitecture',
+    'UtilityMixin',
+    'Autoencoder',
+    'Decoder',
+    'Encoder',
+    'load_arch',
+]
+
+_optional_imports: dict[str, set[str]] = {'flow': {'NormFlow', 'NormFlowEncoder'}}
+
+try:
+    from netloader.architectures.flows import NormFlow, NormFlowEncoder
+    __all__.extend(_optional_imports['flow'])
+except (ModuleNotFoundError, ImportError) as e:
+    pass
+
+
+def __getattr__(name: str) -> Any:
+    if name == 'BaseNetwork':
+        warn(
+            'BaseNetwork is deprecated, please use BaseArchitecture instead',
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        return BaseArchitecture
+    if name == 'load_net':
+        warn(
+            'load_net is deprecated, please use load_arch instead',
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        return load_arch
+    if name not in __all__ and name in _optional_imports['flow']:
+        raise ImportError(f'Cannot import {name}, normalising flow architectures require the '
+                          f'package Zuko')
+    raise AttributeError(f"module {__name__} has no attribute {name}")