Feature/#186

.github/dependabot.yml

@@ -0,0 +1,18 @@
+version: 2
+updates:
+- package-ecosystem: pip
+  directory: "/"
+  schedule:
+    interval: daily
+    time: "20:00"
+  open-pull-requests-limit: 10
+  ignore:
+  - dependency-name: black
+    versions:
+    - 21.4b0
+  - dependency-name: mxnet
+    versions:
+    - 1.8.0
+  - dependency-name: mypy
+    versions:
+    - "0.800"

.github/workflows/ci-pytest-linux.yaml

@@ -33,13 +33,15 @@ jobs:
         run: |
           poetry run pip install -U pip
           poetry run pip install -U torch torchvision
+          poetry run pip install tensorflow tensorflow-datasets mxnet
           poetry install --no-interaction
 
       - name: Install Dependencies with Comet.ml
         if: "matrix.comet == true"
         run: |
           poetry run pip install -U pip
           poetry run pip install -U torch torchvision
+          poetry run pip install tensorflow tensorflow-datasets mxnet
           poetry install --no-interaction --extras comet
 
       - name: Lint with flake8

.github/workflows/ci-pytest-mac.yaml

@@ -33,13 +33,15 @@ jobs:
         run: |
           poetry run pip install -U pip
           poetry run pip install -U torch torchvision
+          poetry run pip install tensorflow tensorflow-datasets mxnet
           poetry install --no-interaction
 
       - name: Install Dependencies with Comet.ml
         if: "matrix.comet == true"
         run: |
           poetry run pip install -U pip
           poetry run pip install -U torch torchvision
+          poetry run pip install tensorflow tensorflow-datasets mxnet
           poetry install --no-interaction --extras comet
 
       - name: Lint with flake8

docker/Aarch64.Dockerfile

@@ -0,0 +1,11 @@
+FROM khirotaka0122/torch:1.8.0-aarch64-py3.8.8
+
+SHELL ["/bin/bash", "-c"]
+
+WORKDIR /enchanter-build
+COPY enchanter/ /enchanter-build/enchanter/
+COPY README.md pyproject.toml /enchanter-build/
+
+RUN poetry install --no-interaction --no-ansi
+
+WORKDIR /workspace

enchanter/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.9.0a0"
+__version__ = "0.9.0b0"
 __author__ = "Hirotaka Kawashima"
 __author_email__ = "khirotaka@vivaldi.net"
 __license__ = "Apache-2.0"

enchanter/addons/layers/dense_interpolation.py

@@ -9,7 +9,7 @@
 
 import numpy as np
 from torch import tensor, bmm
-from torch.tensor import Tensor
+from torch import Tensor
 from torch.nn import Module
 
 

enchanter/addons/layers/mlp.py

@@ -9,7 +9,7 @@
 
 from typing import List, Callable, Union
 from torch import relu
-from torch.tensor import Tensor
+from torch import Tensor
 from torch.nn import Module, ModuleList, Linear, Sequential, Conv1d, ReLU
 
 

enchanter/addons/layers/se_layer.py

@@ -1,4 +1,4 @@
-from torch.tensor import Tensor
+from torch import Tensor
 from torch import nn
 
 

enchanter/engine/runner.py

@@ -21,7 +21,7 @@
 
 import torch
 from torch.nn import Module
-from torch.tensor import Tensor
+from torch import Tensor
 from torch.cuda import is_available, amp
 from torch.optim.optimizer import Optimizer
 from torch.utils.data import DataLoader, SubsetRandomSampler

enchanter/preprocessing/signal/functions.py

@@ -147,6 +147,7 @@ def adjust_sequences(
                 ``fill=["ffill" or int or float]``
 
         dtype:  Specify the data type of NumPy. The data type of the output series is determined based on this value.
+                ``dtype`` must be a float.
 
     Examples:
         >>> import numpy as np

enchanter/tasks/regression.py

@@ -12,7 +12,7 @@
 import numpy as np
 from sklearn.metrics import r2_score
 import torch
-from torch.nn.modules import Module
+from torch.nn import Module
 from torch.nn.modules.loss import _Loss
 from torch.optim.optimizer import Optimizer
 from torch.cuda.amp import GradScaler, autocast

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "enchanter"
-version = "0.9.0a0"
+version = "0.9.0b0"
 description = "Enchanter is a library for machine learning tasks for comet.ml users."
 authors = ["Hirotaka Kawashima <khirotaka@vivaldi.net>"]
 license = "Apache-2.0"
@@ -24,12 +24,9 @@ netron = { version = "^4.5", optional = true }
 comet_ml = { version = "^3.1.3", optional = true }
 
 [tool.poetry.dev-dependencies]
-mxnet = "1.7.0.post2"
 pytest = "^6.0"
 flake8 = "^3.8"
 black = "^20.8b1"
-tensorflow = { version = "^2.3.0", allow-prereleases = true }
-tensorflow-datasets = "^4.0"
 sphinx-material = "^0.0.32"
 sphinx-copybutton = "^0.3"
 sphinx-autoapi = "^1.5"

tests/preprocessing/signal/test_functions.py

@@ -63,8 +63,8 @@ def test_adjust_sequences_4():
         np.random.randn(100, 5),
     ]
 
-    new = signal.adjust_sequences(x, dtype=np.int64)
-    assert new.dtype == np.int64
+    new = signal.adjust_sequences(x, dtype=np.float32)
+    assert new.dtype == np.float32
 
 
 def test_adjust_sequences_5():