From 53f64f536478e950c317af5d239b42a90d5e9a53 Mon Sep 17 00:00:00 2001 From: s0nicboOm Date: Tue, 14 Nov 2023 10:36:30 -0800 Subject: [PATCH] fix: add docstrings Signed-off-by: s0nicboOm --- numalogic/config/_config.py | 2 ++ numalogic/udfs/tools.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/numalogic/config/_config.py b/numalogic/config/_config.py index a0946bae..42abc89b 100644 --- a/numalogic/config/_config.py +++ b/numalogic/config/_config.py @@ -85,6 +85,8 @@ class LightningTrainerConf: @dataclass class TrainerConf: + """Schema for defining the trainer config.""" + train_hours: int = 24 * 8 # 8 days worth of data min_train_size: int = 2000 retrain_freq_hr: int = 24 diff --git a/numalogic/udfs/tools.py b/numalogic/udfs/tools.py index 0a46d828..78105b97 100644 --- a/numalogic/udfs/tools.py +++ b/numalogic/udfs/tools.py @@ -18,7 +18,7 @@ class _DedupMetadata(NamedTuple): - """namedtuple for Dedup Metadata.""" + """Data Structure for Dedup Metadata.""" msg_read_ts: Optional[str] msg_train_ts: Optional[str]