join.py: Add FeatureCollection to _FeatureCollectionType.

PiperOrigin-RevId: 659548085
google · Aug 5, 2024 · d1a124a · d1a124a
1 parent 0449c55
commit d1a124a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/python/ee/join.py b/python/ee/join.py
@@ -8,11 +8,13 @@
 from ee import computedobject
 from ee import ee_string
 from ee import featurecollection
-from ee import filter
+from ee import filter as ee_filter
 
 _EeBoolType = Union[Any, computedobject.ComputedObject]
-_FeatureCollectionType = Union[Any, computedobject.ComputedObject]
-_FilterType = Union[filter.Filter, computedobject.ComputedObject]
+_FeatureCollectionType = Union[
+    Any, featurecollection.FeatureCollection, computedobject.ComputedObject
+]
+_FilterType = Union[ee_filter.Filter, computedobject.ComputedObject]
 _StringType = Union[str, ee_string.String, computedobject.ComputedObject]