Commit 8ada75a2 authored by Paulo Medeiros's avatar Paulo Medeiros
Browse files

Fix retrieval of DTG metadata in save_df_as_obsoul

parent ecbbc608
......@@ -75,7 +75,9 @@ def save_df_as_obsoul(df, fpath=None, export_params=None):
<http://www.rclace.eu/File/Data_Assimilation/2007/lace_obspp.pdf>
Args:
df (pandas.Dataframe): Input dataframe.
df (netatmoqc.load_data.DataFrameWithDtgMetadata): Input dataframe.
This type is inherited from pandas.DataFrame, but it has added
DTG metadata info (df.metadata_dict["dtg"]).
fpath (pathlib.Path): Output file path (Default value = OBSOUL${DTG}).
export_params (list): Parameters to save in the OBSOULD file.
(Default value=["pressure","temperature","humidity","sum_rain_1"])
......@@ -84,7 +86,7 @@ def save_df_as_obsoul(df, fpath=None, export_params=None):
if export_params is None:
export_params = ["pressure", "temperature", "humidity", "sum_rain_1"]
dtg = df.obs_dtg
dtg = df.metadata_dict["dtg"]
# Define n_params, the "number of bodies" in obsoul lingo
df_data_cols = [_ for _ in export_params if _ in df.columns]
......
......@@ -7,7 +7,7 @@
[tool.poetry]
name = "netatmoqc"
version = "0.3.4.dev0"
version = "0.3.4.dev1"
description = "Use machine learning clustering methods to perform quality control over NetAtmo data"
authors = [
"Paulo V. C. Medeiros <paulo.medeiros@smhi.se>"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment