Fix

qubvel · qubvel · commit 67581c3f18e9 · 2024-05-30T23:35:52.000Z
diff --git a/segmentation_models_pytorch/__init__.py b/segmentation_models_pytorch/__init__.py
@@ -63,6 +63,7 @@ def create_model(
 
 
 __all__ = [
+    "datasets",
     "encoders",
     "decoders",
     "losses",
