rename cloud_type to flag in decode_sky_type

NREL · Oct 17, 2024 · 4199910 · 4199910
1 parent 58dacb1
commit 4199910
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlclouds/p_fun.py b/mlclouds/p_fun.py
@@ -36,6 +36,7 @@ def decode_sky_type(cloud_type):
     """Decode integer cloud types as strings."""
     if 'cloud_type' in cloud_type:
         df = cloud_type.copy()
+        df = df.rename({'cloud_type': 'flag'}, axis=1)
     else:
         df = pd.DataFrame(columns=['flag'], data=cloud_type)
     df.loc[df['flag'] == 0, 'flag'] = 'clear'