Upload 2 files

script.py

@@ -192,32 +192,32 @@ if __name__ == "__main__":
 
 	print('------------ Now you can do your solution ---------------')
 	solution = []
-	from concurrent.futures import ProcessPoolExecutor
-	with ProcessPoolExecutor(max_workers=1) as pool:
-		results = []
-		for i, sample in enumerate(tqdm(dataset)):
-			results.append(pool.submit(predict, sample, visualize=False))
+	# from concurrent.futures import ProcessPoolExecutor
+	# with ProcessPoolExecutor(max_workers=1) as pool:
+	# 	results = []
+	# 	for i, sample in enumerate(tqdm(dataset)):
+	# 		results.append(pool.submit(predict, sample, visualize=False))
 		
-		for i, result in enumerate(tqdm(results)):
-			key, pred_vertices, pred_edges = result.result()
-			solution.append({
-							'__key__': key,
-							'wf_vertices': pred_vertices.tolist(),
-							'wf_edges': pred_edges
-						})
+	# 	for i, result in enumerate(tqdm(results)):
+	# 		key, pred_vertices, pred_edges = result.result()
+	# 		solution.append({
+	# 						'__key__': key,
+	# 						'wf_vertices': pred_vertices.tolist(),
+	# 						'wf_edges': pred_edges
+	# 					})
 	####### added for removing multiprocessing ########
-	# for i, sample in enumerate(tqdm(dataset)):
-	# 	key, pred_vertices, pred_edges = predict(sample, visualize=False)
-	# 	solution.append({
-	# 					'__key__': key,
-	# 					'wf_vertices': pred_vertices.tolist(),
-	# 					'wf_edges': pred_edges
-	# 				})
+	for i, sample in enumerate(tqdm(dataset)):
+		key, pred_vertices, pred_edges = predict(sample, visualize=False)
+		solution.append({
+						'__key__': key,
+						'wf_vertices': pred_vertices.tolist(),
+						'wf_edges': pred_edges
+					})
 	####### added for removing multiprocessing ########
-			if i % 10 == 0:
-				# incrementally save the results in case we run out of time
-				print(f"Processed {i} samples")
-				save_submission(solution, Path(params['output_path']) / "submission.parquet")
+		if i % 2 == 0:
+			# incrementally save the results in case we run out of time
+			print(f"Processed {i} samples")
+			save_submission(solution, Path(params['output_path']) / "submission.parquet")
 	print('------------ Saving results ---------------')
 	save_submission(solution, Path(params['output_path']) / "submission.parquet")
 	print("------------ Done ------------ ")