fix mypy checks

Author: Joseph Hamman

.pre-commit-config.yaml

@@ -33,3 +33,16 @@ repos:
     hooks:
       - id: prettier
         language_version: system
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v0.910
+    hooks:
+      - id: mypy
+        additional_dependencies: [
+            # Type stubs
+            types-setuptools,
+            types-pkg_resources,
+            # Dependencies that are typed
+            numpy,
+            xarray,
+          ]

conftest.py

@@ -1,3 +1,4 @@
+# type: ignore
 import pytest
 
 

doc/conf.py

@@ -12,6 +12,8 @@
 # All configuration values have a default; values that are commented out
 # serve to show the default.
 
+# type: ignore
+
 import os
 import sys
 

setup.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# type: ignore
 import os
 
 from setuptools import find_packages, setup

xbatcher/generators.py

@@ -2,8 +2,7 @@
 
 import itertools
 from collections import OrderedDict
-from collections.abc import Iterator
-from typing import Any, Dict, Hashable
+from typing import Any, Dict, Hashable, Iterator
 
 import xarray as xr
 
@@ -163,7 +162,7 @@ def _gen_batches(self) -> dict:
                 ]
                 dsc = xr.concat(all_dsets, dim='input_batch')
                 new_input_dims = [
-                    dim + new_dim_suffix for dim in self.input_dims
+                    str(dim) + new_dim_suffix for dim in self.input_dims
                 ]
                 batches.append(_maybe_stack_batch_dims(dsc, new_input_dims))
             else:

xbatcher/loaders/torch.py

@@ -57,7 +57,6 @@ def __getitem__(self, idx) -> Tuple[Any, Any]:
 
         if self.target_transform:
             y_batch = self.target_transform(y_batch)
-        print('x_batch.shape', X_batch.shape)
         return X_batch, y_batch