Commit 109c3755 authored by Paulo Medeiros's avatar Paulo Medeiros
Browse files

Make sure remove_irregular_stations returns df, df

To simplify the code and fix an unpacking error in the csv2obsoul
command.
parent 16623341
......@@ -496,8 +496,11 @@ def remove_irregular_stations(
Default value = True)
Returns:
'pandas.Dataframe': Copy of input df, but with the
irregular. stations removed.
{
'pandas.Dataframe': Copy of input df, but with the
irregular. stations removed,
'pandas.Dataframe': Dataframe containing removed movind stations.
}
"""
dtg = df.metadata_dict["dtg"]
......@@ -512,6 +515,7 @@ def remove_irregular_stations(
dtg,
)
else:
df_moving_stations = pd.DataFrame.empty
logger.warning("Not checking for moving stations")
if duplicates:
......@@ -526,4 +530,4 @@ def remove_irregular_stations(
df.metadata_dict["dtg"] = dtg
return (df, df_moving_stations) if moving else df
return df, df_moving_stations
......@@ -7,7 +7,7 @@
[tool.poetry]
name = "netatmoqc"
version = "0.3.4.dev3"
version = "0.3.4.dev4"
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