Add additional type annotation

materialsproject · Sep 16, 2019 · 809b657 · 809b657
1 parent a3522e2
commit 809b657
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymatgen/analysis/magnetism/jahnteller.py b/pymatgen/analysis/magnetism/jahnteller.py
@@ -16,7 +16,7 @@
 from pymatgen.symmetry.analyzer import SpacegroupAnalyzer
 import warnings
 
-from typing import Dict, Tuple, Union, Optional
+from typing import Dict, Tuple, Union, Optional, Any
 
 MODULE_DIR = os.path.dirname(os.path.abspath(__file__))
 
@@ -228,7 +228,7 @@ def trim(f):
 
         # perform aggregation of all sites
         if jt_sites:
-            analysis = {"active": True}
+            analysis = {"active": True}  # type: Dict[str, Any]
             # if any site could exhibit 'strong' Jahn-Teller effect
             # then mark whole structure as strong
             strong_magnitudes = [site["strength"] == "strong" for site in jt_sites]