tensorneko 0.3.15__py3-none-any.whl → 0.3.16__py3-none-any.whl

tensorneko/__init__.py

@@ -1,7 +1,6 @@
 import os.path
 
 from . import backend
-from . import callback
 from . import dataset
 from . import debug
 from . import evaluation
@@ -14,14 +13,11 @@ from . import preprocess
 from . import util
 from . import visualization
 from .io import read, write
-from .neko_model import NekoModel
 from .neko_module import NekoModule
-from .neko_trainer import NekoTrainer
 
 __version__ = io.read.text(os.path.join(util.get_tensorneko_path(), "version.txt"))
 
 __all__ = [
-    "callback",
     "dataset",
     "backend",
     "evaluation",
@@ -34,11 +30,28 @@ __all__ = [
     "util",
     "visualization",
     "debug",
-    "NekoModel",
-    "NekoTrainer",
     "NekoModule",
     "read",
     "write",
 ]
 
+try:
+    from . import callback
+except ImportError:
+    pass
+else:
+    __all__.append("callback")
 
+try:
+    from .neko_model import NekoModel
+except ImportError:
+    pass
+else:
+    __all__.append("NekoModel")
+
+try:
+    from .neko_trainer import NekoTrainer
+except ImportError:
+    pass
+else:
+    __all__.append("NekoTrainer")

tensorneko/version.txt

@@ -1 +1 @@
-0.3.15
+0.3.16

{tensorneko-0.3.15.dist-info → tensorneko-0.3.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko
-Version: 0.3.15
+Version: 0.3.16
 Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
@@ -23,19 +23,19 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Utilities
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: torch >=1.9.0
-Requires-Dist: torchaudio >=0.9.0
-Requires-Dist: torchvision >=0.10.0
-Requires-Dist: torchmetrics >=0.7.3
-Requires-Dist: tensorboard >=2.0.0
-Requires-Dist: lightning <2.2,>=2.0
-Requires-Dist: pillow >=8.1
-Requires-Dist: av >=8.0.3
-Requires-Dist: numpy >=1.20.1
-Requires-Dist: einops >=0.3.0
-Requires-Dist: tensorneko-util ==0.3.15
-Requires-Dist: pysoundfile >=0.9.0 ; platform_system == "Windows"
+Requires-Dist: av (>=8.0.3)
+Requires-Dist: einops (>=0.3.0)
+Requires-Dist: numpy (>=1.20.1)
+Requires-Dist: pillow (>=8.1)
+Requires-Dist: tensorneko-util (==0.3.16)
+Requires-Dist: torch (>=1.9.0)
+Requires-Dist: torchaudio (>=0.9.0)
+Requires-Dist: torchmetrics (>=0.7.3)
+Requires-Dist: torchvision (>=0.10.0)
+Requires-Dist: pysoundfile (>=0.9.0) ; platform_system == "Windows"
+Provides-Extra: lightning
+Requires-Dist: lightning (<2.2,>=2.0) ; extra == 'lightning'
+Requires-Dist: tensorboard (>=2.0.0) ; extra == 'lightning'
 
 <h1 style="text-align: center">TensorNeko</h1> 
 
@@ -64,8 +64,11 @@ Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lig
 
 ## Install
 
+The tensorneko requires pytorch and pytorch-lightning (optional), and you can install it with below command.
+
 ```shell
-pip install tensorneko
+pip install tensorneko  # for PyTorch only
+pip install tensorneko[lightning]  # for PyTorch and Lightning
 ```
 
 To use the library without PyTorch and PyTorch Lightning, you can install the util library (support Python 3.7 ~ 3.12 with limited features) with following command.
@@ -273,7 +276,7 @@ t0 = time.time()
 with Server(view, port=8000):
     for i, data in enumerate(data_list):
         preprocessing(data) # do some processing here
-        
+
         x = time.time() - t0  # time since the start of the program
         y = i # processed number of data
         line_chart.add(x, y)  # add to the line chart
@@ -494,7 +497,7 @@ def process_data(n: int):
         return n
     else:
         return None
-    
+
 
 data = get_data()
 data = data.map(process_data).get_or_else(-1)  # if the response is None, return -1

{tensorneko-0.3.15.dist-info → tensorneko-0.3.16.dist-info}/RECORD

@@ -1,8 +1,8 @@
-tensorneko/__init__.py,sha256=lLVC4StQ5q9OFJTceNrJbj-CIFHFMRvFQyGYJTjuRM4,812
+tensorneko/__init__.py,sha256=uh1HNn1sNpX1bbOqAE_kNJfrH4eMtEzus0hO-Fh9tEw,990
 tensorneko/neko_model.py,sha256=QTbdOAg9ki0ix6mDL_Qu8Wmd5WJOoUFF3M1SXEp3KGc,10551
 tensorneko/neko_module.py,sha256=qELXvguSjWo_NvcRQibiFl0Qauzd9JWLSnT4dbGNS3Y,1473
 tensorneko/neko_trainer.py,sha256=JC8qoKSZ5ngz3grf3S0SjvIFVktDIP_GExth5aFfbGA,10074
-tensorneko/version.txt,sha256=G9xsooAOFSfDHEuC_syU1UK-FLNztcrz3G6L4qv-6vM,6
+tensorneko/version.txt,sha256=XSXl8BxVFKqB3N-DasVRdbWcnpx7WRAyDr5dG_Eo7DA,6
 tensorneko/arch/__init__.py,sha256=w4lTUeyBIZelrnSjlBFWUF0erzOmBFl9FqeWQuSOyKs,248
 tensorneko/arch/auto_encoder.py,sha256=j6PWWyaNYaYNtw_zZ9ikzhCASqe9viXR3JGBIXSK92Y,2137
 tensorneko/arch/binary_classifier.py,sha256=1MkEbReXKLdDksRG5Rsife40grJk08EVDcNKp54Xvb4,2316
@@ -85,8 +85,8 @@ tensorneko/visualization/log_graph.py,sha256=NvOwWVc_petXWYdgaHosPFLa43sHBeacbYc
 tensorneko/visualization/matplotlib.py,sha256=xs9Ssc44ojZX65QU8-fftA7Ug_pBuZ3TBtM8vETNq9w,1568
 tensorneko/visualization/image_browser/__init__.py,sha256=AtykhAE3bXQS6SOWbeYFeeUE9ts9XOFMvrL31z0LoMg,63
 tensorneko/visualization/watcher/__init__.py,sha256=Nq752qIYvfRUZ8VctKQRSqhxh5KmFbWcqPfZlijVx6s,379
-tensorneko-0.3.15.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
-tensorneko-0.3.15.dist-info/METADATA,sha256=ul7vopVVtLMuxQjnlc6L6srzKooifZT_yOJYqPR9Di8,19743
-tensorneko-0.3.15.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-tensorneko-0.3.15.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
-tensorneko-0.3.15.dist-info/RECORD,,
+tensorneko-0.3.16.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
+tensorneko-0.3.16.dist-info/METADATA,sha256=Y2dgqx89FzcdEV8hkGrFeTNeCmdJ-YqbYWej5WxRSAg,19998
+tensorneko-0.3.16.dist-info/WHEEL,sha256=g4nMs7d-Xl9-xC9XovUrsDHGXt-FT0E17Yqo92DEfvY,92
+tensorneko-0.3.16.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
+tensorneko-0.3.16.dist-info/RECORD,,

{tensorneko-0.3.15.dist-info → tensorneko-0.3.16.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: bdist_wheel (0.34.2)
 Root-Is-Purelib: true
 Tag: py3-none-any