Version 0.1.0

spfrommer · Nov 13, 2023 · 059fe61 · 059fe61
1 parent d062d8b
commit 059fe61
Show file tree

Hide file tree

Showing 8 changed files with 14 additions and 8 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1,5 @@
+include torchexplorer/*.py
+include torchexplorer/vega/vega_dataless.json
+include torchexplorer/standalone/README.md
+include torchexplorer/standalone/data/.keep
+include torchexplorer/standalone/templates/index.html
diff --git a/examples/cifar10resnet/main.py b/examples/cifar10resnet/main.py
@@ -12,7 +12,7 @@
 from pytorch_lightning.loggers import WandbLogger
 from sklearn import datasets
 
-from torchexplorer.api import watch, LIGHTNING_EPOCHS
+from torchexplorer import watch, LIGHTNING_EPOCHS
 
 from data.cifar10_datamodule import CIFAR10DataModule
 from data.metadata import get_normalize_layer

diff --git a/examples/irismlp/main.py b/examples/irismlp/main.py
@@ -10,7 +10,7 @@
 
 import click
 
-from torchexplorer.api import watch, LIGHTNING_EPOCHS
+from torchexplorer import watch, LIGHTNING_EPOCHS
 
 
 wandb_init_params = {

diff --git a/setup.py b/setup.py
@@ -1,18 +1,19 @@
-from setuptools import setup, find_namespace_packages
+from setuptools import setup, find_namespace_packages, find_packages
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setup(
     name='torchexplorer',
-    version='0.0.1',
+    version='0.1.0',
     description="Interactively inspect pytorch modules during training.",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/spfrommer/torchexplorer",
     author="Samuel Pfrommer",
     author_email="sam.pfrommer@gmail.com",
-    packages=find_namespace_packages(include=['torchexplorer.*']),
+    packages=find_packages(),
+    include_package_data=True,
     install_requires=[
         'click',
         'colorama',

diff --git a/tests/integration/local.py b/tests/integration/local.py
@@ -4,7 +4,7 @@
 import torch
 from torch import nn
 
-from torchexplorer.api import StructureWrapper, watch
+from torchexplorer import StructureWrapper, watch
 from torchexplorer.core import (
     ModuleInvocationHistograms, ModuleInvocationStructure
 )

diff --git a/tests/integration/remote.py b/tests/integration/remote.py
@@ -7,7 +7,7 @@
 import torch
 from torch import nn
 
-from torchexplorer.api import watch
+from torchexplorer import watch
 
 import infra
 

diff --git a/torchexplorer/__init__.py b/torchexplorer/__init__.py
@@ -1 +1 @@
-from torchexplorer.api import watch, LIGHTNING_EPOCHS
+from torchexplorer.api import watch, LIGHTNING_EPOCHS, StructureWrapper
diff --git a/torchexplorer/standalone/data/.keep b/torchexplorer/standalone/data/.keep
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from torchexplorer.api import watch, LIGHTNING_EPOCHS
		from torchexplorer.api import watch, LIGHTNING_EPOCHS, StructureWrapper