Update

[ghstack-poisoned]

Author: Vincent Moens

setup.py

@@ -11,7 +11,7 @@
 import sys
 from pathlib import Path
 
-from setuptools import Extension, setup
+from setuptools import Extension, find_packages, setup
 from setuptools.command.build_ext import build_ext
 
 ROOT_DIR = Path(__file__).parent.resolve()
@@ -92,4 +92,7 @@ def get_extensions():
         "build_ext": CMakeBuild,
         "clean": clean,
     },
+    packages=find_packages(
+        exclude=("test", "tutorials", "packaging", "gallery", "docs")
+    ),
 )

test/smoke_test.py

@@ -5,7 +5,16 @@
 
 
 def test_imports():
+    print("Importing numpy")  # noqa
+    import numpy  # noqa
+
+    print("Importing torch")  # noqa
+    import torch  # noqa
+
+    print("Importing tensordict")  # noqa
     from tensordict import TensorDict  # noqa: F401
+
+    print("Importing tensordict nn")  # noqa
     from tensordict.nn import TensorDictModule  # noqa: F401
 
     # # Check that distributed is not imported