remove setup, add 3.6,3.7,3.8 to travis

.travis.yml

@@ -2,22 +2,21 @@ branches:
   only:
   - master
   - stable
+  - dev
 
 # Set build language to Python
 language: python
 
-# Set python version to 3.7
-python: 3.7
+python:
+  - "3.6"
+  - "3.7"
+  - "3.8"
 
+before_install:
+- pip install poetry
 install:
-  - python setup.py install
-  - pip install codecov
-  - pip install pytest pytest-cov
-  - pip install numpy==1.17.2
-  - pip install torch
-  - pip install torchvision
+  - poetry install
 script:
-  - chmod +x setup.py
   - pytest --cov=./ --cov-report xml
 
 # Report results to DeepSource

dvc.yaml

@@ -1,11 +1,11 @@
 stages:
   sanity-check:
-    cmd: python run_sanity_check.py
+    cmd: python sanity_check/run_sanity_check.py
     deps:
-      - GaborNet/GaborLayers.py
       - GaborNet/GaborLayer.py
       - data
-      - run_sanity_check.py
+      - sanity_check/run_sanity_check.py
+      - sanity_check/dataset.py
     params:
       - sanity_check.epoch
     metrics:

pyproject.toml

@@ -1,18 +1,35 @@
 [tool.poetry]
-name = "gabornet"
+name = "GaborNet"
 version = "0.2.0"
-description = ""
+description = "Meet Gabor Layer"
+repository = "https://github.com/iKintosh/GaborNet"
+readme = "README.md"
 authors = ["an.alekseev <[email protected]>"]
 license = "MIT"
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3",
+    "Topic :: Software Development :: Libraries :: Python Modules",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Education",
+    "License :: OSI Approved :: MIT License",
+    "Environment :: GPU :: NVIDIA CUDA :: 10.2",
+    "Operating System :: Unix"
+]
+packages = [
+    { include = "GaborNet" }
+]
 
 [tool.poetry.dependencies]
-python = "^3.8.0"
-pytest = "^6.0.1"
+python = "^3.6.1"
 numpy = "^1.19.1"
 torch = "^1.6.0"
 torchvision = "^0.7.0"
 
 [tool.poetry.dev-dependencies]
+pytest = "^6.0.1"
 pep257 = "^0.7.0"
 pylint = "^2.6.0"
 pycodestyle = "^2.6.0"