From 2fb538e24eeae03783fceb236bb9b53bbec3652e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Tue, 24 Jan 2023 13:01:26 +0000
Subject: [PATCH] typing(signal): cleanup and add typing
---
src/agora/io/signal.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/src/agora/io/signal.py b/src/agora/io/signal.py
index 4a822757..c0c3e840 100644
--- a/src/agora/io/signal.py
+++ b/src/agora/io/signal.py
@@ -256,10 +256,10 @@ class Signal(BridgeH5):
def get_raw(
self,
- dataset: str,
+ dataset: str or t.List[str],
in_minutes: bool = True,
lineage: bool = False,
- ) -> pd.DataFrame:
+ ) -> pd.DataFrame or t.List[pd.DataFrame]:
"""
Load data from a h5 file and return as a dataframe.
@@ -345,7 +345,10 @@ class Signal(BridgeH5):
fullname: str,
node: t.Union[h5py.Dataset, h5py.Group],
):
- """Store the name of a signal if it is a leaf node (a group with no more groups inside) and if it starts with extraction."""
+ """
+ Store the name of a signal if it is a leaf node
+ (a group with no more groups inside) and if it starts with extraction.
+ """
if isinstance(node, h5py.Group) and np.all(
[isinstance(x, h5py.Dataset) for x in node.values()]
):
--
GitLab