Fix filtering by int64-type identifiers, see GH-7

earthobservations · Apr 9, 2023 · cc24080 · cc24080
1 parent ba8b745
commit cc24080
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -5,6 +5,7 @@ phenodata changelog
 development
 ===========
 - Fix ``nearest-station`` with ``--format=json``
+- Fix filtering by ``int64``-type identifiers, see GH-7
 
 2020-12-29 0.11.0
 =================

diff --git a/phenodata/dwd/pheno.py b/phenodata/dwd/pheno.py
@@ -372,7 +372,8 @@ def flux(self, results, criteria=None):
                 continue
             reference = results[field]
             if key in criteria and criteria[key]:
-                expression &= reference.isin(criteria[key])
+                values = map(int, criteria[key])
+                expression &= reference.isin(values)
 
         # Apply filter expression to DataFrame
         if type(expression) is not bool: