Wip (#50)

* small tweaks
achaiah · Oct 8, 2021 · b974480 · b974480
1 parent ce5a2a2
commit b974480
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 9 deletions.
diff --git a/docs/source/README.md b/docs/source/README.md
@@ -7,7 +7,7 @@
 [![pypi](https://img.shields.io/pypi/v/pywick.svg)](https://pypi.org/project/pywick/)
 [![python compatibility](https://img.shields.io/pypi/pyversions/pywick.svg)](https://pywick.readthedocs.io/en/latest/)
 [![license](https://img.shields.io/pypi/l/pywick.svg)](https://github.com/achaiah/pywick/blob/master/LICENSE.txt)
-
+[![Documentation Status](https://readthedocs.org/projects/pywick/badge/?version=latest)](https://pywick.readthedocs.io/en/latest/?badge=latest) 
 </div>
 
 #### High-Level Training framework for Pytorch

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -40,8 +40,8 @@
 # The short X.Y version
 version = ''
 # The full version, including alpha/beta/rc tags
-release = '0.6.5'
 
+release = '0.6.0'
 
 # -- General configuration ---------------------------------------------------
 
@@ -179,9 +179,14 @@
 
 # Exclude imports
 autodoc_mock_imports = [
+    "albumentations",
     "h5py",
-    "torch",
-    "torchvision",
     "numpy",
-    "pandas"
+    "opencv-python-headless",
+    "pandas",
+    "pycm",
+    "scikit-image",
+    "scipy",
+    "torch",
+    "torchvision"
 ]
diff --git a/docs/source/segmentation_guide.md b/docs/source/segmentation_guide.md
@@ -2,7 +2,6 @@
 
 In a short while we will publish a walk-through that will go into detail
 on how to do segmentation with Pywick. In the meantime, if you feel
-adventurous feel free to look at our
-[README](https://github.com/achaiah/pywick/blob/master/README.md).
+adventurous feel free to look at our [README](https://pywick.readthedocs.io/en/latest/README.html).
 
-You can also take a look at our [Classification guide](https://github.com/achaiah/pywick/blob/master/docs/source/classification_guide.md) to get a good idea of how to get started on your own. The segmentation training process is very similar but involves more complicated directory structure for data.
+You can also take a look at our [Classification guide](https://pywick.readthedocs.io/en/latest/classification_guide.html) to get a good idea of how to get started on your own. The segmentation training process is very similar but involves more complicated directory structure for data.
diff --git a/pywick/models/classification/dpn/dualpath.py b/pywick/models/classification/dpn/dualpath.py
@@ -228,7 +228,7 @@ def __init__(
             self.key_stride = 2
             self.has_proj = True
         else:
-            if block_type is not 'normal':
+            if block_type != 'normal':
                 raise AssertionError
             self.key_stride = 1
             self.has_proj = False