TYP: add type hints for enzo

yt-project · Oct 8, 2021 · bb537d0 · bb537d0
1 parent bdf2609
commit bb537d0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/yt/frontends/enzo/io.py b/yt/frontends/enzo/io.py
@@ -1,3 +1,5 @@
+from typing import Dict
+
 import numpy as np
 
 from yt.geometry.selection_routines import GridSelector
@@ -7,7 +9,7 @@
 
 _convert_mass = ("particle_mass", "mass")
 
-_particle_position_names = {}
+_particle_position_names: Dict[str, str] = {}
 
 
 class IOHandlerPackedHDF5(BaseIOHandler):