fix(cli): fix cli chanel close

gnes-ai · Aug 30, 2019 · f1402f5 · f1402f5
1 parent d441f58
commit f1402f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_pipeline_train.py b/tests/test_pipeline_train.py
@@ -22,6 +22,7 @@ def test_train(self):
         self.assertRaises(RuntimeError, de.encode, 1)
         de.train()
         self.assertEqual(2, de.encode(1))
+        self.dirname = os.path.dirname(__file__)
 
     def tearDown(self):
         if os.path.exists('dummy-pipeline.bin'):
@@ -42,7 +43,7 @@ def test_pipeline_train(self):
         self.assertEqual(4, a.encode(1))
 
     def test_load_yaml(self):
-        p = BaseEncoder.load_yaml('yaml/pipeline-multi-encoder.yml')
+        p = BaseEncoder.load_yaml(os.path.join(self.dirname, 'yaml', 'pipeline-multi-encoder.yml'))
         self.assertRaises(RuntimeError, p.encode, 1)
         p.train(1)
         self.assertEqual(5, p.encode(1))