isort

Author: SkafteNicki

requirements.txt

@@ -1,2 +1,2 @@
 torch>=1.3
-matplotlib>=3.3.3
+matplotlib>=3.1.3

setup.py

@@ -1,29 +1,16 @@
 #!/usr/bin/env python
-# Copyright The GeoML Team
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
 import os
 from io import open
 
-from setuptools import setup, find_packages, Command
+from setuptools import Command, find_packages, setup
 
 try:
     import builtins
 except ImportError:
     import __builtin__ as builtins
 
 PATH_ROOT = os.path.dirname(__file__)
-builtins.__STOCHMAN_SETUP__ = True
+builtins.__TORCHPLOT_SETUP__ = True
 
 import torchplot
 
@@ -43,28 +30,27 @@ def run(self):
         os.system("rm -vrf ./build ./dist ./*.pyc ./*.tgz ./*.egg-info")
 
 
-PATH_ROOT = os.path.dirname(__file__)
+with open("README.md", encoding="utf-8") as f:
+    long_description = f.read()
 
 
-def load_readme(path_dir=PATH_ROOT):
-    with open(os.path.join(path_dir, "README.md"), encoding="utf-8") as f:
-        long_description = f.read()
-    return long_description
+with open("requirements.txt", "r") as reqs:
+    requirements = reqs.read().split()
 
 
 setup(
     name="torchplot",
     version=torchplot.__version__,
     description=torchplot.__docs__,
-    long_description=load_readme(PATH_ROOT),
+    long_description=long_description,
     long_description_content_type="text/markdown",
     author=torchplot.__author__,
     author_email=torchplot.__author_email__,
     license=torchplot.__license__,
     packages=find_packages(exclude=["tests", "tests/*"]),
     python_requires=">=3",
-    install_requires=["torch", "matplotlib"],
-    download_url="https://github.com/CenterBioML/torchplot/archive/0.1.4.zip",
+    install_requires=requirements,
+    download_url="https://github.com/CenterBioML/torchplot/archive/0.1.5.zip",
     classifiers=[
         "Environment :: Console",
         "Natural Language :: English",

tests/test_torchplot.py

@@ -11,15 +11,15 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import pytest
-from inspect import getmembers, isfunction
-from collections import namedtuple
 import string
+from collections import namedtuple
+from inspect import getmembers, isfunction
 
-import torch
+import matplotlib.pyplot as plt
 import numpy as np
+import pytest
+import torch
 
-import matplotlib.pyplot as plt
 import torchplot as tp
 
 Inputs = namedtuple("case", ["x", "y"])

torchplot/__init__.py

@@ -1,22 +1,10 @@
-# Copyright The GeoML Team
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+#!/usr/bin/env python
 """Root package info."""
 import os
 import time
 
 _this_year = time.strftime("%Y")
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 __author__ = "Nicki Skafte Detlefsen et al."
 __author_email__ = "[email protected]"
 __license__ = "Apache-2.0"

torchplot/core.py

@@ -1,20 +1,9 @@
 #!/usr/bin/env python
-# Copyright The GeoML Team
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+from inspect import getdoc, getmembers, isfunction
+
 import matplotlib.pyplot as plt
 import torch
-from inspect import getmembers, isfunction, getdoc
+
 
 # Function to convert a list of arguments containing torch tensors, into
 # a corresponding list of arguments containing numpy arrays