Minor changes

Author: marshka

README.md

@@ -4,15 +4,11 @@
     <h3>Neural spatiotemporal forecasting with PyTorch</h3>
     <hr>
     <p>
-    <a href='https://pypi.org/project/torch-spatiotemporal/'>
-        <img alt="PyPI" src="https://img.shields.io/pypi/v/torch-spatiotemporal">
-    </a>
+    <a href='https://pypi.org/project/torch-spatiotemporal/'><img alt="PyPI" src="https://img.shields.io/pypi/v/torch-spatiotemporal"></a>
     <img alt="PyPI - Python Version" src="https://img.shields.io/badge/python-%3E%3D3.8-blue">
     <!-- img alt="PyPI - Python Version" src="https://img.shields.io/pypi/pyversions/torch-spatiotemporal" -->
     <img alt="Total downloads" src="https://static.pepy.tech/badge/torch-spatiotemporal">
-    <a href='https://torch-spatiotemporal.readthedocs.io/en/latest/?badge=latest'>
-        <img src='https://readthedocs.org/projects/torch-spatiotemporal/badge/?version=latest' alt='Documentation Status' />
-    </a>
+    <a href='https://torch-spatiotemporal.readthedocs.io/en/latest/?badge=latest'><img src='https://readthedocs.org/projects/torch-spatiotemporal/badge/?version=latest' alt='Documentation Status' /></a>
     </p>
     <p>
     🚀 <a href="https://torch-spatiotemporal.readthedocs.io/en/latest/usage/quickstart.html">Getting Started</a> - 📚 <a href="https://torch-spatiotemporal.readthedocs.io/en/latest/">Documentation</a> - 💻 <a href="https://torch-spatiotemporal.readthedocs.io/en/latest/notebooks/a_gentle_introduction_to_tsl.html">Introductory notebook</a>

tsl/engines/predictor.py

@@ -127,7 +127,7 @@ def load_model(self, filename: str):
             assert model_cls == self.model_cls
             if model_kwargs is not None:
                 for k, v in model_kwargs.items():
-                    assert v == self.model_kwargs[k]
+                    assert v == self.model_kwargs[k], f'{v}'
         else:
             logger.warning("Predictor with already instantiated model is "
                            f"loading a state_dict from {filename}. Cannot "
@@ -157,8 +157,8 @@ def _filter_forward_kwargs(self, kwargs: dict) -> dict:
             model_args = self._model_fwd_signature['signature']
             filtered = set(kwargs).difference(model_args)
             forwarded = set(kwargs).intersection(model_args)
-            msg = f"Only args {list(forwarded)} are forwarded to the model " \
-                  f"({self.model.__class__.__name__}). "
+            msg = (f"Only args {list(forwarded)} are forwarded to the model "
+                   f"({self.model.__class__.__name__}).")
             if len(filtered):
                 msg = f"Arguments {list(filtered)} are filtered out. " + msg
             logger.warning(msg)