Update src/get_labels_from_samples.py

src/get_labels_from_samples.py

@@ -1,14 +1,14 @@
-from segments import SegmentsClient
 import math
+from segments import SegmentsClient
 
 def get_samples(client, dataset_identifier):
     dataset = client.get_dataset(dataset_identifier)
     samples_count = getattr(dataset, "samples_count", 0)
-    roundup = round((samples_count / 1000), 0)
+    total_pages = math.ceil(samples_count / 1000)
     page = 1
     per_page = 1000
     samples = []
-    while page <= roundup:
+    while page <= total_pages:
         response = client.get_samples(dataset_identifier, per_page=per_page, page=page)
         samples.extend(response)
         page += 1
@@ -103,4 +103,4 @@ def main():
     print(all_sensor_frames_and_annotations)
 
 if __name__ == "__main__":
-    main()
+    main()