From 2ea95ede0a5ef26ed6d58c9982f823bf51d75da9 Mon Sep 17 00:00:00 2001 From: paulzierep Date: Thu, 25 Jan 2024 12:19:03 +0100 Subject: [PATCH] mypy stats_utils --- data/usage_stats/stats_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data/usage_stats/stats_utils.py b/data/usage_stats/stats_utils.py index bdcde1b5..a651e9e9 100644 --- a/data/usage_stats/stats_utils.py +++ b/data/usage_stats/stats_utils.py @@ -10,10 +10,10 @@ tool_stats_path = project_path.joinpath(project_path, "tool_usage.csv") tool_stats_df = pd.read_csv(tool_stats_path, sep=",") -tool_stats_df.columns = ["tool_name", "count"] +tool_stats_df.columns = pd.Index(["tool_name", "count"]) -def remove_version(val): +def remove_version(val: str) -> str: if "/" in val: split_val = val.split("/") val = "/".join(split_val[:-1])