tensorneko 0.3.14__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/msg/__init__.py

@@ -0,0 +1,3 @@
+from tensorneko_util.msg import push_gotify
+
+__all__ = ["push_gotify"]

tensorneko/version.txt

@@ -1 +1 @@
-0.3.14
+0.3.16

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko
-Version: 0.3.14
+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
@@ -8,13 +8,14 @@ Author-email: smczx@hotmail.com
 License: UNKNOWN
 Project-URL: Bug Tracker, https://github.com/ControlNet/tensorneko/issues
 Project-URL: Source Code, https://github.com/ControlNet/tensorneko
-Keywords: deep learning,pytorch,AI
+Keywords: deep learning,pytorch,AI,data processing
 Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Intended Audience :: Developers
@@ -22,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.14
-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> 
 
@@ -63,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.
@@ -72,11 +76,18 @@ To use the library without PyTorch and PyTorch Lightning, you can install the ut
 pip install tensorneko_util
 ```
 
-Some cpu bound functions are implemented by `pyo3`, and you can install the optimized version with below command.
+Some cpu bound functions are implemented by rust-based `pyo3`, and you can install the optimized version with below command.
 ```shell
 pip install tensorneko_lib
 ```
 
+Some CLI tools are provided in the `tensorneko_tool` package, and you can install it with below command.
+```shell
+pipx install tensorneko_tool  # or `pip install tensorneko_tool`
+```
+
+Then you can use the CLI tools `tensorneko` in the terminal.
+
 ## Neko Layers, Modules and Architectures
 
 Build an MLP with linear layers. The activation and normalization will be placed in the hidden layers.
@@ -265,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
@@ -486,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
@@ -519,7 +530,7 @@ This library provides event bus based reactive tools. The API integrates the Pyt
 # useful decorators for default event bus
 from tensorneko.util import subscribe
 # Event base type
-from tensorneko.util import Event
+from tensorneko.util import Event, EventBus
 
 class LogEvent(Event):
     def __init__(self, message: str):
@@ -549,6 +560,7 @@ if __name__ == '__main__':
     # emit an event, and then the event handler will be invoked
     # The sequential order is not guaranteed
     LogEvent("Hello world!")
+    EventBus.default.wait()  # it's not blocking, need to call wait manually before exit.
     # one possible output:
     # Hello world! in another thread
     # Hello world! async
@@ -685,4 +697,14 @@ Functions list (in `tensorneko`):
 - `is_bad_num`
 - `count_parameters`
 
+## TensorNeko Tools
+
+Some CLI tools are provided in the `tensorneko_tool` package.
+
+The `gotify` can send a message to the Gotify server, with the environment variables `GOTIFY_URL` and `GOTIFY_TOKEN` set.
+
+```shell
+tensorneko gotify "Script finished!"
+```
+
 

{tensorneko-0.3.14.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=EE5wqaalqNgsRrFbdyrk4CNfvN0GLaydNnLLB0janbo,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
@@ -63,6 +63,7 @@ tensorneko/module/inception.py,sha256=2p8AjgTIk5wLGC-JnrmXIehh6yNCu2tYc2axjzUTGM
 tensorneko/module/mlp.py,sha256=AFN6xmvlrNWOflLqVl-zVkoOJRZpYxYB4bnI10JG5CU,3361
 tensorneko/module/residual.py,sha256=S59TqiiD_310HQ3a6s3r49XY_7Dc4RGxONQtSvzEfN0,2958
 tensorneko/module/transformer.py,sha256=h4NvH3zGa0rZt0bv6e8VM31SimbQKRcocSR42zJYVoY,7602
+tensorneko/msg/__init__.py,sha256=GHrHjzw__0DcPBHBN6GzHrD8PD_7CwOZPRullOaZyW0,71
 tensorneko/notebook/__init__.py,sha256=Pgz4aTJg5_3zTzBIaML4LHAMOFgrBOHw3o00ZuvNuQU,171
 tensorneko/optim/__init__.py,sha256=89XjYQICij8SkrW5iryfZgmbxcTDxA3hhVZTgR4588o,33
 tensorneko/optim/lr_scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -84,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.14.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
-tensorneko-0.3.14.dist-info/METADATA,sha256=O9XRez7LsZUJM-2qxxRroMJ99l4N3KFpC4FjgQ5KaGA,19061
-tensorneko-0.3.14.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-tensorneko-0.3.14.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
-tensorneko-0.3.14.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.14.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