From 6dc35120ade1147aa69a4251e6be47b90da6473d Mon Sep 17 00:00:00 2001
From: Norbert Walkowiak <norwal2@st.amu.edu.pl>
Date: Tue, 18 Apr 2023 07:56:51 +0200
Subject: [PATCH] fix

---
 s487175-create-dataset-script.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index 055700a..b55c834 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -9,11 +9,16 @@ diamonds = pd.read_csv('diamonds.csv')
 cutoff = os.environ.get('CUTOFF', None)
 if cutoff is None:
     print("CUTOFF parameter not set.")
-else:
-    cutoff = int(cutoff)
-    diamonds = diamonds[:cutoff]
-    diamonds.to_csv('diamonds.csv', index=False) #nadpisanie pliku
     diamonds
+else:
+    if cutoff.strip() == "":
+        print("CUTOFF parameter is empty.")
+        diamonds
+    else:
+        cutoff = int(cutoff)
+        diamonds = diamonds[:cutoff]
+        diamonds.to_csv('diamonds.csv', index=False) ## nadpisanie pliku
+        diamonds
 
 # %%
 #przydzielanie nazwy kolumny z id