Remove pkl files (#116)

* Doesn't generate pkl files anymore

* Increase epochs so test doesn't fail

* Fix lint

* Fix lint

Author: fealho

tests/integration/test_ctgan.py

@@ -9,6 +9,8 @@
 model are not checked.
 """
 
+import tempfile as tf
+
 import numpy as np
 import pandas as pd
 
@@ -120,9 +122,10 @@ def test_save_load():
 
     ctgan = CTGANSynthesizer(epochs=1)
     ctgan.fit(data, discrete_columns)
-    ctgan.save("test_ctgan.pkl")
 
-    ctgan = CTGANSynthesizer.load("test_ctgan.pkl")
+    with tf.TemporaryDirectory() as temporary_directory:
+        ctgan.save(temporary_directory + "test_tvae.pkl")
+        ctgan = CTGANSynthesizer.load(temporary_directory + "test_tvae.pkl")
 
     sampled = ctgan.sample(1000)
     assert set(sampled.columns) == {'continuous', 'discrete'}

tests/integration/test_tvae.py

@@ -9,6 +9,8 @@
 model are not checked.
 """
 
+import tempfile as tf
+
 import numpy as np
 import pandas as pd
 
@@ -70,11 +72,12 @@ def test_save_load():
     })
     discrete_columns = ['discrete']
 
-    tvae = TVAESynthesizer(epochs=1)
+    tvae = TVAESynthesizer(epochs=10)
     tvae.fit(data, discrete_columns)
-    tvae.save("test_tvae.pkl")
 
-    tvae = TVAESynthesizer.load("test_tvae.pkl")
+    with tf.TemporaryDirectory() as temporary_directory:
+        tvae.save(temporary_directory + "test_tvae.pkl")
+        tvae = TVAESynthesizer.load(temporary_directory + "test_tvae.pkl")
 
     sampled = tvae.sample(1000)
     assert set(sampled.columns) == {'continuous', 'discrete'}