add ehrapy lib

batmen-lab · Dec 21, 2023 · 8015f48 · 8015f48
1 parent 5bf2cfd
commit 8015f48
Show file tree

Hide file tree

Showing 10 changed files with 15 additions and 2 deletions.
diff --git a/chatbot_ui_biomania/components/Chat/LibCardSelect.tsx b/chatbot_ui_biomania/components/Chat/LibCardSelect.tsx
@@ -11,11 +11,12 @@ export const libImages: { [key: string]: string } = {
   'biotite': '/apps/biotite.png',
   'deap': '/apps/deap.png',
   'eletoolkit': '/apps/eletoolkit.jpg',
-  'pyopenms': '/apps/pyopenms.png',
+  'pyopenms': '/apps/pyopenms.jpg',
   'scenicplus': '/apps/SCENIC.png',
   'scvi-tools': '/apps/scvitools.svg',
-  'sonata': '/apps/SONATA.png',
+  'sonata': '/apps/SONATA.jpg',
   'MIOSTONE': '/apps/MIOSTONE.jpg',
+  'ehrapy': '/apps/ehrapy.jpg',
   //'custom': '/apps/customize.jpg',
 };
 
@@ -50,6 +51,7 @@ export const LibCardSelect = () => {
       { id: 'scvi-tools', name: 'scvi-tools' },
       { id: 'sonata', name: 'sonata' },
       { id: 'MIOSTONE', name: 'MIOSTONE' },
+      { id: 'ehrapy', name: 'ehrapy' },
       //{ id: 'custom', name: 'custom' },
     ];
 

diff --git a/chatbot_ui_biomania/public/apps/SCENIC.png b/chatbot_ui_biomania/public/apps/SCENIC.png
diff --git a/chatbot_ui_biomania/public/apps/SONATA.jpg b/chatbot_ui_biomania/public/apps/SONATA.jpg
diff --git a/chatbot_ui_biomania/public/apps/SONATA.png b/chatbot_ui_biomania/public/apps/SONATA.png
diff --git a/chatbot_ui_biomania/public/apps/ehrapy.jpg b/chatbot_ui_biomania/public/apps/ehrapy.jpg
diff --git a/chatbot_ui_biomania/public/apps/pyopenms.jpg b/chatbot_ui_biomania/public/apps/pyopenms.jpg
diff --git a/chatbot_ui_biomania/public/apps/pyopenms.png b/chatbot_ui_biomania/public/apps/pyopenms.png
diff --git a/src/configs/Lib_cheatsheet.json b/src/configs/Lib_cheatsheet.json
@@ -169,5 +169,14 @@
         "READTHEDOC_LINK": null,
         "TUTORIAL_HTML_PATH": null,
         "TUTORIAL_GITHUB": null
+    },
+    "ehrapy": {
+        "LIB": "ehrapy",
+        "LIB_ALIAS": "ehrapy",
+        "API_HTML_PATH": "ehrapy.readthedocs.io/en/latest/usage/usage.html",
+        "GITHUB_LINK": null,
+        "READTHEDOC_LINK": null,
+        "TUTORIAL_HTML_PATH": null,
+        "TUTORIAL_GITHUB": null
     }
 }
diff --git a/src/models/train_retriever.py b/src/models/train_retriever.py
@@ -5,6 +5,7 @@
 import pandas as pd
 from datetime import datetime
 import torch
+#torch.cuda.set_per_process_memory_fraction(0.5)
 import torch.nn as nn
 from sentence_transformers import SentenceTransformer, models, InputExample, losses, LoggingHandler
 from torch.utils.data import DataLoader

diff --git a/src/requirements.txt b/src/requirements.txt
@@ -6,6 +6,7 @@ bitsandbytes==0.41.1
 deap==1.4.1
 deepspeed==0.12.0
 docstring_parser==0.15
+ehrapy==0.6.0
 Flask==3.0.0
 flask_cors==4.0.0
 flask_socketio==5.3.6