Fix cuda error due to wrong tensor initialization with cuda

Author: Jean-KOUAGOU

ontolearn/learners/drill.py

@@ -525,7 +525,7 @@ def learn_from_replay_memory(self) -> None:
  current_state_batch = torch.cat(current_state_batch, dim=0).to(self.device)
  # N, 1, dim
  next_state_batch = torch.cat(next_state_batch, dim=0).to(self.device)
- y = torch.Tensor(y, device=self.device)
+ y = torch.Tensor(y).to(self.device)
 
  try:
  assert current_state_batch.shape[1] == next_state_batch.shape[1] == self.emb_pos.shape[1] == \

ontolearn/utils/__init__.py

@@ -156,7 +156,7 @@ def read_csv(path)->Union[None,pd.DataFrame]: # pragma: no cover
  if assertion_path_isfile(path):
  df = pd.read_csv(path, index_col=0)
  assert (df.all()).all() # all columns and all rows are not none.
- df.index = df.index.map(lambda x: x.replace("<", "").replace(">", ""))
+ df.index = df.index.map(lambda x: str(x).replace("<", "").replace(">", ""))
  return df
  else:
  return None