From 7a96f1e82cb66355970ba94b4328771a56686935 Mon Sep 17 00:00:00 2001
From: Daniel Svitan <daniel.svitan.team7274dev@gmail.com>
Date: Sun, 15 Dec 2024 12:26:58 +0100
Subject: [PATCH] :hammer: Changes sleep enum

---
 CLEAN.md         | 4 ++--
 analyze_sleep.py | 4 ++--
 clean.py         | 6 +++---
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/CLEAN.md b/CLEAN.md
index 666ba7e..cdb17ff 100644
--- a/CLEAN.md
+++ b/CLEAN.md
@@ -66,7 +66,7 @@ The cleaned dataset will have the following structure:
 ### Sleep
 
 ```
-0 - long sleepers
+0 - short sleepers
 1 - medium sleepers
-2 - short sleepers
+2 - long sleepers
 ```
diff --git a/analyze_sleep.py b/analyze_sleep.py
index c9ce9be..589f087 100644
--- a/analyze_sleep.py
+++ b/analyze_sleep.py
@@ -4,9 +4,9 @@ from analyze import analyze
 
 dataset = np.load("clean.npy")
 print(f"dataset shape: {dataset.shape}, analyzing column 10 (sleep)")
-print("\t0 - long sleepers")
+print("\t0 - short sleepers")
 print("\t1 - medium sleepers")
-print("\t2 - short sleepers")
+print("\t2 - long sleepers")
 print("")
 
 
diff --git a/clean.py b/clean.py
index 4cbdc95..34fd331 100644
--- a/clean.py
+++ b/clean.py
@@ -112,11 +112,11 @@ def parse_commute(txt: str) -> int:
 
 def parse_sleep(txt: str) -> int:
     match txt:
-        case "9 hodín a viac":
+        case "6 hodín a menej":
             return 0
         case "7 až 9 hodín":
             return 1
-        case "6 hodín a menej":
+        case "9 hodín a viac":
             return 2
         case _:
             print("ERROR: Coudln't determine sleep")
@@ -132,7 +132,7 @@ def parse_absence(txt: str) -> float:
             txt = input("Please enter fixed value: ")
 
 
-for i in range(1, len(df)):
+for i in range(len(arr)):
     row = arr[i]
     current = []