Spaces:
Sleeping
Sleeping
Make TDC import lazy and add tdc to requirements for Spaces
Browse files- requirements.txt +1 -0
- src/ingest/tdc_loader.py +3 -1
requirements.txt
CHANGED
|
@@ -5,3 +5,4 @@ scikit-learn
|
|
| 5 |
joblib
|
| 6 |
streamlit
|
| 7 |
altair
|
|
|
|
|
|
| 5 |
joblib
|
| 6 |
streamlit
|
| 7 |
altair
|
| 8 |
+
tdc
|
src/ingest/tdc_loader.py
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
from pathlib import Path
|
| 2 |
import pandas as pd
|
| 3 |
-
from tdc.single_pred import ADME
|
| 4 |
|
| 5 |
|
| 6 |
def load_and_save_adme_dataset(name: str, out_dir: str = "data/raw") -> Path:
|
|
@@ -14,6 +13,9 @@ def load_and_save_adme_dataset(name: str, out_dir: str = "data/raw") -> Path:
|
|
| 14 |
out_path = Path(out_dir)
|
| 15 |
out_path.mkdir(parents=True, exist_ok=True)
|
| 16 |
|
|
|
|
|
|
|
|
|
|
| 17 |
data = ADME(name=name)
|
| 18 |
df = data.get_data()
|
| 19 |
|
|
|
|
| 1 |
from pathlib import Path
|
| 2 |
import pandas as pd
|
|
|
|
| 3 |
|
| 4 |
|
| 5 |
def load_and_save_adme_dataset(name: str, out_dir: str = "data/raw") -> Path:
|
|
|
|
| 13 |
out_path = Path(out_dir)
|
| 14 |
out_path.mkdir(parents=True, exist_ok=True)
|
| 15 |
|
| 16 |
+
# Lazy import to avoid import-time failures in deployment environments
|
| 17 |
+
from tdc.single_pred import ADME
|
| 18 |
+
|
| 19 |
data = ADME(name=name)
|
| 20 |
df = data.get_data()
|
| 21 |
|