Small imports update.

Author: LouisTier

torchsom/__init__.py

@@ -2,13 +2,18 @@
 from .decay import DECAY_FUNCTIONS
 from .distances import DISTANCE_FUNCTIONS
 from .neighborhood import NEIGHBORHOOD_FUNCTIONS
+from .plotting import SOMVisualizer, VisualizationConfig
+from .utils import *
 
 __version__ = "0.1.0"
 
-# Define what should be imported when using 'from TorchSOM import *'
+# Define what should be imported when using 'from torchsom import *'
 __all__ = [
     "TorchSOM",
     "DISTANCE_FUNCTIONS",
     "DECAY_FUNCTIONS",
     "NEIGHBORHOOD_FUNCTIONS",
+    "SOMVisualizer",
+    "VisualizationConfig",
+    # No need to list utils methods here, they are already imported
 ]

torchsom/__pycache__/__init__.cpython-311.pyc

@@ -1,9 +1,7 @@
 from dataclasses import dataclass, field
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Dict, List, Optional, Tuple, Union
 
-import matplotlib.cm as cm
-import matplotlib.colors as mcolors
 import matplotlib.pyplot as plt
 import numpy as np
 import torch