change types for distribution

FusRoman · Dec 13, 2023 · 8040a3d · 8040a3d
1 parent 65a11d2
commit 8040a3d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/fink_mm/distribution/distribution.py b/fink_mm/distribution/distribution.py
@@ -16,7 +16,7 @@
 from fink_mm.distribution.apply_filters import apply_filters
 
 from fink_utils.spark import schema_converter
-from pyspark.sql.types import StructType
+from pyspark.sql.types import StructType, StringType
 
 
 def format_mangrove_col(userschema: StructType) -> StructType:
@@ -45,25 +45,25 @@ def format_mangrove_col(userschema: StructType) -> StructType:
                     "metadata": {},
                     "name": "HyperLEDA_name",
                     "nullable": True,
-                    "type": "string",
+                    "type": StringType(),
                 },
                 {
                     "metadata": {},
                     "name": "TwoMASS_name",
                     "nullable": True,
-                    "type": "string",
+                    "type": StringType(),
                 },
                 {
                     "metadata": {},
                     "name": "lum_dist",
                     "nullable": True,
-                    "type": "string",
+                    "type": StringType(),
                 },
                 {
                     "metadata": {},
                     "name": "ang_dist",
                     "nullable": True,
-                    "type": "string",
+                    "type": StringType(),
                 },
             ],
             "type": "struct",