Update dataset.py

Acbarakat · Aug 21, 2023 · 26099bf · 26099bf
1 parent 8f9af4d
commit 26099bf
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/data/dataset.py b/src/data/dataset.py
@@ -1,4 +1,15 @@
 # -*- coding: utf-8 -*-
+"""
+Getting and transforming dataframes/datasets.
+
+Attributes:
+    SRC_DIR (str): Where the src folder is
+    MODEL_DIR (str): Where the models are stored
+
+Todo:
+    * N/A
+
+"""
 import os
 import json
 from typing import Tuple