From 6b7a0783a37d8c3309e7acf65be8644d1c6e0164 Mon Sep 17 00:00:00 2001 From: Edoardo Pasca Date: Thu, 9 May 2019 12:41:24 +0100 Subject: set proper directory on creation --- Wrappers/Python/ccpi/framework/TestData.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Wrappers/Python/ccpi/framework/TestData.py b/Wrappers/Python/ccpi/framework/TestData.py index 61ed4df..7e8362e 100755 --- a/Wrappers/Python/ccpi/framework/TestData.py +++ b/Wrappers/Python/ccpi/framework/TestData.py @@ -21,7 +21,7 @@ class TestData(object): def load(self, which, size=(512,512), scale=(0,1), **kwargs): if which not in [TestData.BOAT, TestData.CAMERA, TestData.PEPPERS]: raise ValueError('Unknown TestData {}.'.format(which)) - tmp = Image.open(os.path.join(data_dir, which)) + tmp = Image.open(os.path.join(self.data_dir, which)) data = numpy.array(tmp.resize(size)) -- cgit v1.2.3