diff --git a/examples/imagenet/train_test.py b/examples/imagenet/train_test.py index 191ee3c80..1ed9170d7 100644 --- a/examples/imagenet/train_test.py +++ b/examples/imagenet/train_test.py @@ -71,7 +71,7 @@ def test_train_and_evaluate(self, model): workdir = tempfile.mkdtemp() # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # Define training configuration diff --git a/examples/lm1b/input_pipeline_test.py b/examples/lm1b/input_pipeline_test.py index f964b0826..ce9331843 100644 --- a/examples/lm1b/input_pipeline_test.py +++ b/examples/lm1b/input_pipeline_test.py @@ -51,7 +51,7 @@ def _get_datasets(self): vocab_path = os.path.join(tempfile.mkdtemp(), 'sentencepiece_model') # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable with tfds.testing.mock_data(num_examples=128, data_dir=data_dir): diff --git a/examples/lm1b/train_test.py b/examples/lm1b/train_test.py index f506005ac..8db21ca67 100644 --- a/examples/lm1b/train_test.py +++ b/examples/lm1b/train_test.py @@ -60,7 +60,7 @@ def test_train_and_evaluate(self): workdir = tempfile.mkdtemp() # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable with tfds.testing.mock_data(num_examples=128, data_dir=data_dir): diff --git a/examples/mnist/train_test.py b/examples/mnist/train_test.py index fecc2f36c..b443a7fc4 100644 --- a/examples/mnist/train_test.py +++ b/examples/mnist/train_test.py @@ -60,7 +60,7 @@ def test_train_and_evaluate(self): workdir = tempfile.mkdtemp() # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + "/.tfds/metadata" # pylint: disable=unused-variable # Define training configuration. diff --git a/examples/ogbg_molpcba/train_test.py b/examples/ogbg_molpcba/train_test.py index e8200bf9d..c6f6fe1ad 100644 --- a/examples/ogbg_molpcba/train_test.py +++ b/examples/ogbg_molpcba/train_test.py @@ -339,7 +339,7 @@ def test_train_and_evaluate(self): workdir = tempfile.mkdtemp() # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable # Get the test configuration. diff --git a/examples/sst2/input_pipeline_test.py b/examples/sst2/input_pipeline_test.py index 0b916aa5d..48de2f70c 100644 --- a/examples/sst2/input_pipeline_test.py +++ b/examples/sst2/input_pipeline_test.py @@ -48,7 +48,7 @@ def _get_vocab_path(self): def _get_dataset(self, vocab_path): """Uses mock data to create the dataset.""" # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable with tfds.testing.mock_data(num_examples=128, data_dir=data_dir): return input_pipeline.TextDataset(vocab_path=vocab_path, split='train') diff --git a/examples/wmt/input_pipeline_test.py b/examples/wmt/input_pipeline_test.py index 4657dda8d..68e71f586 100644 --- a/examples/wmt/input_pipeline_test.py +++ b/examples/wmt/input_pipeline_test.py @@ -51,7 +51,7 @@ def _get_datasets(self): vocab_path = os.path.join(tempfile.mkdtemp(), 'sentencepiece_model') # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable with tfds.testing.mock_data(num_examples=128, data_dir=data_dir): diff --git a/examples/wmt/train_test.py b/examples/wmt/train_test.py index 5b233e680..5a5f8ae66 100644 --- a/examples/wmt/train_test.py +++ b/examples/wmt/train_test.py @@ -60,7 +60,7 @@ def test_train_and_evaluate(self): workdir = tempfile.mkdtemp() # Go two directories up to the root of the flax directory. - flax_root_dir = pathlib.Path(__file__).parents[2] + flax_root_dir = pathlib.Path(__file__).absolute().parents[2] data_dir = str(flax_root_dir) + '/.tfds/metadata' # pylint: disable=unused-variable with tfds.testing.mock_data(num_examples=128, data_dir=data_dir):