Skip to content
Snippets Groups Projects
Commit 4a1e0b5c authored by Alexandros-Nikolaos Ziogas's avatar Alexandros-Nikolaos Ziogas
Browse files

Fixed typing.

parent c75b98c2
Branches
......@@ -6,7 +6,7 @@ import numpy as np
import argparse
import json
from pathlib import Path
from typing import Tuple
from typing import List, Tuple
def read_json_data(folder: str = '.', frmwrk: str = 'dace_cpu') -> pd.DataFrame:
......@@ -20,7 +20,7 @@ def read_json_data(folder: str = '.', frmwrk: str = 'dace_cpu') -> pd.DataFrame:
df = pd.concat([df, data])
return df
def filter_nodes(df: pd.DataFrame, allowed: list[int]) -> pd.DataFrame:
def filter_nodes(df: pd.DataFrame, allowed: List[int]) -> pd.DataFrame:
return df[df.nodes.isin(allowed)]
......
......@@ -4,7 +4,7 @@ import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import argparse
from typing import Tuple
from typing import List, Tuple
def read_csv_data(folder: str = '.', frmwrk: str = 'dace_cpu') -> pd.DataFrame:
......@@ -14,7 +14,7 @@ def read_csv_data(folder: str = '.', frmwrk: str = 'dace_cpu') -> pd.DataFrame:
df = pd.concat([df, data])
return df
def filter_nodes(df: pd.DataFrame, allowed: list[int]) -> pd.DataFrame:
def filter_nodes(df: pd.DataFrame, allowed: List[int]) -> pd.DataFrame:
return df[df.nodes.isin(allowed)]
......
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