Changed data path to data
Browse files- agent/agent.py +3 -2
agent/agent.py
CHANGED
|
@@ -16,8 +16,9 @@ class SigSpace(Basic_Agent):
|
|
| 16 |
self.conversation = []
|
| 17 |
self.conversation.append({"role": "system", "content": self.system_prompt})
|
| 18 |
|
| 19 |
-
# initialize data
|
| 20 |
-
path = pathlib.Path("/home/ubuntu/giovanni/code/Tahoe_Hackathon/datasets")
|
|
|
|
| 21 |
# jump_path = pathlib.Path("/home/ubuntu/giovanni/data")
|
| 22 |
self.jump_tahoe_drug_metadata = pd.read_csv(path/"drug_metadata_inchikey.csv")
|
| 23 |
self.jump_similarity_score = pd.read_csv(path/"compound_genetic_perturbation_cosine_similarity_inchikey.csv")
|
|
|
|
| 16 |
self.conversation = []
|
| 17 |
self.conversation.append({"role": "system", "content": self.system_prompt})
|
| 18 |
|
| 19 |
+
# initialize data path
|
| 20 |
+
# path = pathlib.Path("/home/ubuntu/giovanni/code/Tahoe_Hackathon/datasets") # on lambda
|
| 21 |
+
path = pathlib.Path("data")
|
| 22 |
# jump_path = pathlib.Path("/home/ubuntu/giovanni/data")
|
| 23 |
self.jump_tahoe_drug_metadata = pd.read_csv(path/"drug_metadata_inchikey.csv")
|
| 24 |
self.jump_similarity_score = pd.read_csv(path/"compound_genetic_perturbation_cosine_similarity_inchikey.csv")
|