Commit e9d04d06 authored by alat-rights's avatar alat-rights
Browse files

make test_bert_tokenizer use absolute path

parent a69f9428
import pytest
import os
from os.path import join
@pytest.mark.torch
......@@ -35,6 +37,8 @@ def test_loading():
loader = FASTALoader(
featurizer=featurizer, legacy=False, auto_add_annotations=True)
data = loader.create_dataset(input_files="./data/uniprot_truncated.fasta")
cwd = os.getcwd()
data = loader.create_dataset(
input_files=join(cwd, "data/uniprot_truncated.fasta"))
assert data.X.shape == (61, 3, 5)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment