Use another criterion to switch to simulation mode in stream2raw

astrolabsoftware · Jan 29, 2025 · 719e577 · 719e577
1 parent c563e8b
commit 719e577
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/bin/ztf/stream2raw.py b/bin/ztf/stream2raw.py
@@ -93,9 +93,8 @@ def main():
         alert_schema, _, alert_schema_json = get_schemas_from_avro(args.schema)
 
     # Decode the Avro data, and keep only (timestamp, data)
-    if args.producer == "sims":
-        # using custom from_avro (not available for Spark 2.4.x)
-        # it will be available from Spark 3.0 though
+    if args.servers.startswith("localhost"):
+        # Only for test suite
         df_decoded = df.select([
             from_avro(df["value"], alert_schema_json).alias("decoded")
         ])