Mocking more modules and fixed typo in documentation.

muammar · Mar 13, 2020 · 602f987 · 602f987
1 parent 51c4145
commit 602f987
Show file tree

Hide file tree

Showing 5 changed files with 32 additions and 27 deletions.
diff --git a/docs/environment.yml b/docs/environment.yml
@@ -14,7 +14,4 @@ dependencies:
         - sphinxcontrib-inlinesyntaxhighlight
         - ase
         - m2r
-        - joblib
-        - msgpack
-        - msgpack_numpy
 
diff --git a/docs/requirements.txt b/docs/requirements.txt
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -15,20 +15,42 @@
 
 import os
 import sys
+
 sys.path.insert(0, os.path.abspath("../../"))
 
 try:
     import torch
 except ModuleNotFoundError:
     from unittest.mock import MagicMock
 
-
     class Mock(MagicMock):
-       @classmethod
-       def __getattr__(cls, name):
-           return MagicMock()
-
-    MOCK_MODULES = ['torch', "torch.optim", "torch.nn.Module", "plotly"]
+        @classmethod
+        def __getattr__(cls, name):
+            return MagicMock()
+
+    MOCK_MODULES = [
+        "torch",
+        "torch.optim",
+        "torch.nn.Module",
+        "plotly",
+        "joblib",
+        "msgpack",
+        "msgpack_numpy",
+        "seaborn",
+        "dask_ml",
+        "ase",
+        "scipy",
+        "numpy",
+        "pip",
+        "pandas",
+        "plotly",
+        "pytest",
+        "matplotlib",
+        "dscribe",
+        "msgpack_python",
+        "scikit_learn",
+        "dask"
+    ]
 
     sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 

diff --git a/docs/source/hpc.rst b/docs/source/hpc.rst
@@ -29,7 +29,7 @@ carry out computations. This is useful when prototyping and building your
 pipeline withouth wasting time waiting for HPC resources in a crowded cluster
 facility.
 
-ML4Chem can run with:code:`LocalCluster` objects, for which the scripts have
+ML4Chem can run with :code:`LocalCluster` objects, for which the scripts have
 to contain the following::
 
    from dask.distributed import Client, LocalCluster

diff --git a/ml4chem/atomistic/potentials.py b/ml4chem/atomistic/potentials.py
@@ -118,7 +118,9 @@ def load(Cls, model=None, params=None, preprocessor=None, **kwargs):
                 weights = load(model)
                 # TODO remove after de/serialization is fixed.
                 try:
-                    weights = {key.decode("utf-8"): value for key, value in weights.items()}
+                    weights = {
+                        key.decode("utf-8"): value for key, value in weights.items()
+                    }
                 except AttributeError:
                     weights = {key: value for key, value in weights.items()}
 
@@ -412,4 +414,3 @@ def calculate(self, atoms, properties, system_changes):
 
             # Populate ASE's self.results dict
             self.results["energy"] = energy
-