Replacing atomicwrites with os.write (#353)

* replace AtomicWriter with os.write

* remove atomicwrites from requirements

* Delete a.txt

* Delete default-936efe33-f225-4038-a9d9-986ca34c3e6c.ipynb

* Delete default-2ada21bd-5e38-4de5-a2ac-7ef3be7f920e.ipynb

* atomic writer

Co-authored-by: Juan Daniel Torres <[email protected]>

Author: juandaanieel

adaptive/utils.py

@@ -7,7 +7,6 @@
 from itertools import product
 
 import cloudpickle
-from atomicwrites import AtomicWriter
 
 
 def named_product(**items):
@@ -51,8 +50,23 @@ def save(fname, data, compress=True):
     if compress:
         blob = gzip.compress(blob)
 
-    with AtomicWriter(fname, "wb", overwrite=True).open() as f:
-        f.write(blob)
+    temp_file = f"{fname}.{os.getpid()}"
+
+    try:
+        with open(temp_file, "wb") as f:
+            f.write(blob)
+    except OSError:
+        return False
+
+    try:
+        os.replace(temp_file, fname)
+    except OSError:
+        return False
+    finally:
+        if os.path.exists(temp_file):
+            os.remove(temp_file)
+
+    return True
 
 
 def load(fname, compress=True):

docs/environment.yml

@@ -15,7 +15,6 @@ dependencies:
   - plotly=5.3.1
   - ipywidgets=7.6.5
   - jupyter-sphinx=0.3.2
-  - atomicwrites=1.4.0
   - sphinx_fontawesome=0.0.6
   - sphinx=4.2.0
   - m2r2=0.3.1

environment.yml

@@ -19,4 +19,3 @@ dependencies:
   - scikit-optimize>=0.8.1
   - scikit-learn<=0.24.2  # https://github.com/scikit-optimize/scikit-optimize/issues/1059
   - plotly
-  - atomicwrites

setup.py

@@ -28,7 +28,6 @@ def get_version_and_cmdclass(package_name):
     "scipy",
     "sortedcollections >= 1.1",
     "sortedcontainers >= 2.0",
-    "atomicwrites",
     "cloudpickle",
     "loky >= 2.9",
 ]