small change in file

curieo-org · Mar 27, 2024 · eceedb7 · eceedb7
1 parent 4850390
commit eceedb7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/backend/app/make_it_rain.py b/backend/app/make_it_rain.py
@@ -7,9 +7,8 @@
 df = pd.read_csv("chebml_eval_data.csv")
 
 #%%
-df_filter = df[df['Source'] == "Clicnical Trials"].sample(n = 10)
 chembl_data_debug = pd.DataFrame()
-for i in tqdm(df_filter.iterrows()): 
+for i in tqdm(df.iterrows()): 
     url = f"http://127.0.0.1:8000/search?query={i[1]['Question']}"
     data_store = {
         "question": i[1]['Question'],