Fixes

Author: riibeirogabriel

learning_orchestra_client/observe/observe.py

@@ -3,6 +3,7 @@
 
 class Observer:
     __TIMEOUT_TIME_MULTIPLICATION = 1000
+    __MAX_WAIT_TIME = 240
 
     def __init__(self, cluster_ip: str):
         cluster_ip = cluster_ip.replace("http://", "")
@@ -22,6 +23,7 @@ def wait(self, name: str, timeout: int = None) -> dict:
         wait its finish with a
         wait method call.
         timeout: the maximum time to wait the observed step, in seconds.
+        Default time is 4 minutes.
 
         :return: If True it returns a String. Otherwise, it returns
         a dictionary with the content of a mongo collection, representing
@@ -44,6 +46,9 @@ def wait(self, name: str, timeout: int = None) -> dict:
                     ]
             }}
         ]
+
+        timeout = self.__MAX_WAIT_TIME if timeout is None else timeout 
+
         return dataset_collection.watch(
             observer_query,
             full_document='updateLookup',

pipeline/titanic.py

@@ -8,11 +8,11 @@
 dataset_csv = DatasetCsv(CLUSTER_IP)
 
 dataset_csv.insert_dataset_async(
-    url="https://filebin.net/boniydu54k710l54/train.csv?t=s350xryf",
+    url="https://drive.google.com/uc?export=download&id=1ccWBTDolhD50NaIIo50H4k_odvD_rlFr",
     dataset_name="titanic_training",
 )
 dataset_csv.insert_dataset_async(
-    url="https://filebin.net/udtf7eogfgasqnx5/test.csv?t=h79pcy0l",
+    url="https://drive.google.com/uc?export=download&id=1DG5biYaUH-MDhtCfDpZgiSn5Is1ZkMuv",
     dataset_name="titanic_testing"
 )