From fe7103e8862b8d2a07292fac94e37869cf27d6a2 Mon Sep 17 00:00:00 2001
From: Ubuntu
 <azureuser@DiskANN-highmem.20na4qgmoozedmm50zw0432jed.phxx.internal.cloudapp.net>
Date: Wed, 20 Nov 2024 16:22:49 +0000
Subject: [PATCH] added gt urls, added configurable max points

---
 .../runbooks/gen_expiration_time_runbook.py   | 24 ++++++++++++-------
 requirements_py3.10.txt                       |  2 +-
 2 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/neurips23/runbooks/gen_expiration_time_runbook.py b/neurips23/runbooks/gen_expiration_time_runbook.py
index 23ee7ad2..f9b3be65 100644
--- a/neurips23/runbooks/gen_expiration_time_runbook.py
+++ b/neurips23/runbooks/gen_expiration_time_runbook.py
@@ -11,8 +11,10 @@
 timesteps: how long to wait before deleting for each ratio
 seed: seed given to random generator
 do_replace: whether to include replace in runbook or not
+do_delete: whether to include delete in the runbook or not
+max_pts: whether to set a max_pts higher than the max that occurs in the runbook (default None)
 '''
-def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ratios, timesteps, seed = 0, do_replace = False, gt_url = None, do_delete = True):
+def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ratios, timesteps, seed = 0, do_replace = False, gt_url = None, do_delete = True, max_points = None):
     random.seed(seed)
     data = {dataset_name: {}}
 
@@ -99,7 +101,10 @@ def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ra
         }
         t+=1
 
-    data[dataset_name]["max_pts"]=max_num_points
+    if max_points != None & max_points >= max_num_points:
+        data[dataset_name]["max_pts"]=max_points
+    else:
+        data[dataset_name]["max_pts"]=max_num_points
 
     if gt_url is not None:
         data[dataset_name]["gt_url"] = gt_url
@@ -183,8 +188,9 @@ def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ra
 dataset_name = 'msturingpq-1M'
 dataset_size = 1000000
 max_t = 100
-gt_url = None
-gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, False, gt_url)
+gt_url = "https://comp21storage.z5.web.core.windows.net/comp21/MSFT-TURING-ANNS/msturingpq-1M_expiration_time_runbook.yaml"
+max_points = 450000
+gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, False, gt_url, max_points)
 
 ratios = (0, 4, 18)
 timesteps = (0, 100, 20)
@@ -193,8 +199,9 @@ def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ra
 dataset_name = 'msturing-1M'
 dataset_size = 1000000
 max_t = 100
-gt_url = None
-gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, True, gt_url, False)
+gt_url = "https://comp21storage.z5.web.core.windows.net/comp21/MSFT-TURING-ANNS/msturingpq-1M_expiration_time_replace_only_runbook.yaml"
+max_points = None
+gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, True, gt_url, False, max_points)
 
 ratios = (1, 8, 18)
 timesteps = (0, 100, 20)
@@ -203,6 +210,7 @@ def gen_exp_time_runbook(dataset_name, dataset_size, max_t, runbook_filename, ra
 dataset_name = 'msturingpq-1M'
 dataset_size = 1000000
 max_t = 100
-gt_url = None
-gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, True, gt_url)
+gt_url = "https://comp21storage.z5.web.core.windows.net/comp21/MSFT-TURING-ANNS/msturingpq-1M_expiration_time_replace_delete_runbook.yaml"
+max_points = 270000
+gen_exp_time_runbook(dataset_name, dataset_size, max_t, dataset_file, ratios, timesteps, seed, True, gt_url, max_points)
 
diff --git a/requirements_py3.10.txt b/requirements_py3.10.txt
index 3c5b5490..6e451c69 100644
--- a/requirements_py3.10.txt
+++ b/requirements_py3.10.txt
@@ -1,7 +1,7 @@
 ansicolors==1.1.8
 docker==7.1.0
 h5py==3.10.0
-matplotlib==3.3.4
+
 numpy==1.24.2
 pyyaml==6.0
 psutil==5.9.4