ribesstefano commited on
Commit
9bd496b
·
1 Parent(s): f3d4b52

Fixed bug when loading cell2embeddings from wrong directory

Browse files
protac_degradation_predictor/protac_degradation_predictor.py CHANGED
@@ -36,7 +36,7 @@ def get_protac_active_proba(
36
  Dict[str, np.ndarray]: The predictions of the model.
37
  """
38
  protein2embedding = load_protein2embedding()
39
- cell2embedding = load_cell2embedding('data/cell2embedding.pkl')
40
 
41
  # Setup default embeddings
42
  default_protein_emb = np.zeros(config.protein_embedding_size)
 
36
  Dict[str, np.ndarray]: The predictions of the model.
37
  """
38
  protein2embedding = load_protein2embedding()
39
+ cell2embedding = load_cell2embedding()
40
 
41
  # Setup default embeddings
42
  default_protein_emb = np.zeros(config.protein_embedding_size)