Skip to content
This repository was archived by the owner on Aug 31, 2021. It is now read-only.

Commit 68b9c8a

Browse files
committed
Changed from using general tmp folder to per-test tmp folder.
1 parent e27aa15 commit 68b9c8a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

skflow/tests/test_saver.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
class SaverTest(tf.test.TestCase):
2727

2828
def testIris(self):
29-
path = '/tmp/tmp.saver'
29+
path = tf.test.get_temp_dir() + '/tmp.saver'
3030
random.seed(42)
3131
iris = datasets.load_iris()
3232
classifier = skflow.TensorFlowLinearClassifier(n_classes=3)
@@ -38,7 +38,7 @@ def testIris(self):
3838
self.assertGreater(score, 0.5, "Failed with score = {0}".format(score))
3939

4040
def testCustomModel(self):
41-
path = '/tmp/tmp.saver2'
41+
path = tf.test.get_temp_dir() + '/tmp.saver2'
4242
random.seed(42)
4343
iris = datasets.load_iris()
4444
def custom_model(X, y):
@@ -53,7 +53,7 @@ def custom_model(X, y):
5353
self.assertGreater(score, 0.5, "Failed with score = {0}".format(score))
5454

5555
def testDNN(self):
56-
path = 'tmp_saver3'
56+
path = tf.test.get_temp_dir() + '/tmp_saver3'
5757
random.seed(42)
5858
iris = datasets.load_iris()
5959
classifier = skflow.TensorFlowDNNClassifier(hidden_units=[10, 20, 10], n_classes=3)
@@ -69,7 +69,7 @@ def testNoFolder(self):
6969
skflow.TensorFlowEstimator.restore('no_model_path')
7070

7171
def testNoCheckpoints(self):
72-
path = '/tmp/tmp.saver4'
72+
path = tf.test.get_temp_dir() + '/tmp/tmp.saver4'
7373
random.seed(42)
7474
iris = datasets.load_iris()
7575
classifier = skflow.TensorFlowDNNClassifier(hidden_units=[10, 20, 10], n_classes=3)

0 commit comments

Comments
 (0)