improve handler and add requirements.txt
Browse files- handler.py +11 -5
- requirements.txt +28 -0
handler.py
CHANGED
|
@@ -16,11 +16,17 @@ class EndpointHandler():
|
|
| 16 |
|
| 17 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
| 18 |
"""
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
"""
|
| 25 |
smiles = data.pop("inputs")
|
| 26 |
# featurize the smiles
|
|
|
|
| 16 |
|
| 17 |
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
|
| 18 |
"""
|
| 19 |
+
Handle a request to the endpoint
|
| 20 |
+
|
| 21 |
+
Parameters
|
| 22 |
+
----------
|
| 23 |
+
data: dict
|
| 24 |
+
The input data
|
| 25 |
+
|
| 26 |
+
Returns
|
| 27 |
+
-------
|
| 28 |
+
dict
|
| 29 |
+
The output data
|
| 30 |
"""
|
| 31 |
smiles = data.pop("inputs")
|
| 32 |
# featurize the smiles
|
requirements.txt
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
numpy
|
| 2 |
+
pandas
|
| 3 |
+
scikit-learn
|
| 4 |
+
rich
|
| 5 |
+
intake
|
| 6 |
+
joblib
|
| 7 |
+
zarr>=3.0.0
|
| 8 |
+
ipython
|
| 9 |
+
jupyterlab
|
| 10 |
+
rdkit
|
| 11 |
+
molfeat
|
| 12 |
+
datamol
|
| 13 |
+
mordredcommunity
|
| 14 |
+
lightgbm
|
| 15 |
+
seaborn
|
| 16 |
+
splito
|
| 17 |
+
umap-learn
|
| 18 |
+
modal
|
| 19 |
+
chemprop>=2.1.1
|
| 20 |
+
torch
|
| 21 |
+
pytorch-lightning
|
| 22 |
+
wandb
|
| 23 |
+
pytest
|
| 24 |
+
pytest-cov
|
| 25 |
+
codecov
|
| 26 |
+
phx-class-registry
|
| 27 |
+
email-validator
|
| 28 |
+
git+https://github.com/OpenADMET/openadmet_models.git
|