rb757 commited on
Commit
d3a480d
·
1 Parent(s): 2f3094e

Make TDC import lazy and add tdc to requirements for Spaces

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. 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