hmacdope commited on
Commit
f225781
·
1 Parent(s): af1eaad

improve handler and add requirements.txt

Browse files
Files changed (2) hide show
  1. handler.py +11 -5
  2. 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
- data args:
20
- inputs (:obj: `str` | `PIL.Image` | `np.array`)
21
- kwargs
22
- Return:
23
- A :obj:`list` | `dict`: will be serialized and returned
 
 
 
 
 
 
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