@@ -1,5 +1,3 @@

-# -*- coding: utf-8 -*-

-

 import abc

 import functools

 import gzip

@@ -76,10 +76,12 @@ class _RequireAttrsABCMeta(abc.ABCMeta):

         for name, type_ in obj.__annotations__.items():

             try:

                 x = getattr(obj, name)

-                if not isinstance(x, type_):

-                    raise TypeError(

-                        f"The attribute '{name}' is of {type_} instead of {type(x)}."

-                    )

             except AttributeError:

-                raise AttributeError(f"Required attribute {name} not set in __init__.")

+                raise AttributeError(

+                    f"Required attribute {name} not set in __init__."

+                ) from None

+            else:

+                if not isinstance(x, type_):

+                    msg = f"The attribute '{name}' should be of type {type_}, not {type(x)}."

+                    raise TypeError(msg)

         return obj

@@ -80,6 +80,6 @@ class _RequireAttrsABCMeta(abc.ABCMeta):

                     raise TypeError(

                         f"The attribute '{name}' is of {type_} instead of {type(x)}."

                     )

-            except AttributeError as e:

-                raise e(f"Required attribute {name} not set in __init__.")

+            except AttributeError:

+                raise AttributeError(f"Required attribute {name} not set in __init__.")

         return obj

@@ -74,11 +74,12 @@ class _RequireAttrsABCMeta(abc.ABCMeta):

     def __call__(self, *args, **kwargs):

         obj = super().__call__(*args, **kwargs)

         for name, type_ in obj.__annotations__.items():

-            if not hasattr(obj, name):

-                raise ValueError(f"Required attribute {name} not set in __init__.")

-            elif not isinstance(getattr(obj, name), type_):

-                raise TypeError(

-                    f"The attribute '{name}' is of {type_} instead of {type(getattr(obj, name))}"

-                )

-

+            try:

+                x = getattr(obj, name)

+                if not isinstance(x, type_):

+                    raise TypeError(

+                        f"The attribute '{name}' is of {type_} instead of {type(x)}."

+                    )

+            except AttributeError as e:

+                raise e(f"Required attribute {name} not set in __init__.")

         return obj

@@ -71,11 +71,14 @@ def copy_docstring_from(other):

 class _RequireAttrsABCMeta(abc.ABCMeta):

-    required_attributes = []

-

     def __call__(self, *args, **kwargs):

         obj = super().__call__(*args, **kwargs)

-        for name in obj.required_attributes:

+        for name, type_ in obj.__annotations__.items():

             if not hasattr(obj, name):

                 raise ValueError(f"Required attribute {name} not set in __init__.")

+            elif not isinstance(getattr(obj, name), type_):

+                raise TypeError(

+                    f"The attribute '{name}' is of {type_} instead of {type(getattr(obj, name))}"

+                )

+

         return obj

@@ -1,5 +1,6 @@

 # -*- coding: utf-8 -*-

+import abc

 import functools

 import gzip

 import os

@@ -67,3 +68,14 @@ def copy_docstring_from(other):

         return functools.wraps(other)(method)

     return decorator

+

+

+class _RequireAttrsABCMeta(abc.ABCMeta):

+    required_attributes = []

+

+    def __call__(self, *args, **kwargs):

+        obj = super().__call__(*args, **kwargs)

+        for name in obj.required_attributes:

+            if not hasattr(obj, name):

+                raise ValueError(f"Required attribute {name} not set in __init__.")

+        return obj

@@ -7,6 +7,8 @@ import pickle

 from contextlib import contextmanager

 from itertools import product

+from atomicwrites import AtomicWriter

+

 def named_product(**items):

     names = items.keys()

@@ -44,9 +46,13 @@ def save(fname, data, compress=True):

     dirname = os.path.dirname(fname)

     if dirname:

         os.makedirs(dirname, exist_ok=True)

-    _open = gzip.open if compress else open

-    with _open(fname, "wb") as f:

-        pickle.dump(data, f, protocol=pickle.HIGHEST_PROTOCOL)

+

+    blob = pickle.dumps(data, protocol=pickle.HIGHEST_PROTOCOL)

+    if compress:

+        blob = gzip.compress(blob)

+

+    with AtomicWriter(fname, "wb", overwrite=True).open() as f:

+        f.write(blob)

 def load(fname, compress=True):

@@ -4,7 +4,6 @@ import functools

 import gzip

 import os

 import pickle

-import time

 from contextlib import contextmanager

 from itertools import product

@@ -28,13 +27,15 @@ def restore(*learners):

 def cache_latest(f):

     """Cache the latest return value of the function and add it

     as 'self._cache[f.__name__]'."""

+

     @functools.wraps(f)

     def wrapper(*args, **kwargs):

         self = args[0]

-        if not hasattr(self, '_cache'):

+        if not hasattr(self, "_cache"):

             self._cache = {}

         self._cache[f.__name__] = f(*args, **kwargs)

         return self._cache[f.__name__]

+

     return wrapper

@@ -44,18 +45,19 @@ def save(fname, data, compress=True):

     if dirname:

         os.makedirs(dirname, exist_ok=True)

     _open = gzip.open if compress else open

-    with _open(fname, 'wb') as f:

+    with _open(fname, "wb") as f:

         pickle.dump(data, f, protocol=pickle.HIGHEST_PROTOCOL)

 def load(fname, compress=True):

     fname = os.path.expanduser(fname)

     _open = gzip.open if compress else open

-    with _open(fname, 'rb') as f:

+    with _open(fname, "rb") as f:

         return pickle.load(f)

 def copy_docstring_from(other):

     def decorator(method):

         return functools.wraps(other)(method)

+

     return decorator

@@ -1,12 +1,12 @@

 # -*- coding: utf-8 -*-

-from contextlib import contextmanager

 import functools

 import gzip

-from itertools import product

 import os

 import pickle

 import time

+from contextlib import contextmanager

+from itertools import product

 def named_product(**items):

@@ -1,4 +1,5 @@

 # -*- coding: utf-8 -*-

+

 from contextlib import contextmanager

 import functools

 import gzip

@@ -8,12 +8,6 @@ import pickle

 import time

-def timed(f, *args, **kwargs):

-    t_start = time.time()

-    result = f(*args, **kwargs)

-    return result, time.time() - t_start

-

-

 def named_product(**items):

     names = items.keys()

     vals = items.values()

@@ -1,7 +1,10 @@

 # -*- coding: utf-8 -*-

 from contextlib import contextmanager

-from functools import wraps

+import functools

+import gzip

 from itertools import product

+import os

+import pickle

 import time

@@ -30,7 +33,7 @@ def restore(*learners):

 def cache_latest(f):

     """Cache the latest return value of the function and add it

     as 'self._cache[f.__name__]'."""

-    @wraps(f)

+    @functools.wraps(f)

     def wrapper(*args, **kwargs):

         self = args[0]

         if not hasattr(self, '_cache'):

@@ -38,3 +41,26 @@ def cache_latest(f):

         self._cache[f.__name__] = f(*args, **kwargs)

         return self._cache[f.__name__]

     return wrapper

+

+

+def save(fname, data, compress=True):

+    fname = os.path.expanduser(fname)

+    dirname = os.path.dirname(fname)

+    if dirname:

+        os.makedirs(dirname, exist_ok=True)

+    _open = gzip.open if compress else open

+    with _open(fname, 'wb') as f:

+        pickle.dump(data, f, protocol=pickle.HIGHEST_PROTOCOL)

+

+

+def load(fname, compress=True):

+    fname = os.path.expanduser(fname)

+    _open = gzip.open if compress else open

+    with _open(fname, 'rb') as f:

+        return pickle.load(f)

+

+

+def copy_docstring_from(other):

+    def decorator(method):

+        return functools.wraps(other)(method)

+    return decorator

@@ -1,5 +1,6 @@

 # -*- coding: utf-8 -*-

 from contextlib import contextmanager

+from functools import wraps

 from itertools import product

 import time

@@ -24,3 +25,16 @@ def restore(*learners):

     finally:

         for state, learner in zip(states, learners):

             learner.__setstate__(state)

+

+

+def cache_latest(f):

+    """Cache the latest return value of the function and add it

+    as 'self._cache[f.__name__]'."""

+    @wraps(f)

+    def wrapper(*args, **kwargs):

+        self = args[0]

+        if not hasattr(self, '_cache'):

+            self._cache = {}

+        self._cache[f.__name__] = f(*args, **kwargs)

+        return self._cache[f.__name__]

+    return wrapper

@@ -4,14 +4,10 @@ from itertools import product

 import time

-class TimeReturn:

-    def __init__(self, function):

-        self.function = function

-

-    def __call__(self, *args, **kwargs):

-        t_start = time.time()

-        result = self.function(*args, **kwargs)

-        return result, time.time() - t_start

+def timed(f, *args, **kwargs):

+    t_start = time.time()

+    result = f(*args, **kwargs)

+    return result, time.time() - t_start

 def named_product(**items):

@@ -5,21 +5,13 @@ import time

 class TimeReturn:

-    def __init__(self, function, total_time=0):

+    def __init__(self, function):

         self.function = function

-        self.total_time = total_time

     def __call__(self, *args, **kwargs):

         t_start = time.time()

         result = self.function(*args, **kwargs)

-        self.total_time += time.time() - t_start

-        return result, self.total_time

-

-    def __getstate__(self):

-        return (self.function, self.total_time)

-

-    def __setstate__(self, state):

-        self.__init__(*state)

+        return result, time.time() - t_start

 def named_product(**items):

@@ -4,24 +4,6 @@ from itertools import product

 import time

-class WithTime:

-    def __init__(self, function, time=0):

-        self.function = function

-        self.time = time

-

-    def __call__(self, *args, **kwargs):

-        t_start = time.time()

-        result = self.function(*args, **kwargs)

-        self.time += time.time() - t_start

-        return result

-

-    def __getstate__(self):

-        return (self.function, self.time)

-

-    def __setstate__(self, state):

-        self.__init__(*state)

-

-

 class TimeReturn:

     def __init__(self, function, total_time=0):

         self.function = function

@@ -22,26 +22,6 @@ class WithTime:

         self.__init__(*state)

-class AverageTimeReturn:

-    def __init__(self, function, total_time=0, n=0):

-        self.function = function

-        self.total_time = total_time

-        self.n = n

-

-    def __call__(self, *args, **kwargs):

-        t_start = time.time()

-        result = self.function(*args, **kwargs)

-        self.total_time += time.time() - t_start

-        self.n += 1

-        return result, self.total_time / self.n

-

-    def __getstate__(self):

-        return (self.function, self.total_time, self.n)

-

-    def __setstate__(self, state):

-        self.__init__(*state)

-

-

 class TimeReturn:

     def __init__(self, function, total_time=0):

         self.function = function

@@ -42,6 +42,24 @@ class AverageTimeReturn:

         self.__init__(*state)

+class TimeReturn:

+    def __init__(self, function, total_time=0):

+        self.function = function

+        self.total_time = total_time

+

+    def __call__(self, *args, **kwargs):

+        t_start = time.time()

+        result = self.function(*args, **kwargs)

+        self.total_time += time.time() - t_start

+        return result, self.total_time

+

+    def __getstate__(self):

+        return (self.function, self.total_time)

+

+    def __setstate__(self, state):

+        self.__init__(*state)

+

+

 def named_product(**items):

     names = items.keys()

     vals = items.values()

@@ -1,6 +1,45 @@

 # -*- coding: utf-8 -*-

 from contextlib import contextmanager

 from itertools import product

+import time

+

+

+class WithTime:

+    def __init__(self, function, time=0):

+        self.function = function

+        self.time = time

+

+    def __call__(self, *args, **kwargs):

+        t_start = time.time()

+        result = self.function(*args, **kwargs)

+        self.time += time.time() - t_start

+        return result

+

+    def __getstate__(self):

+        return (self.function, self.time)

+

+    def __setstate__(self, state):

+        self.__init__(*state)

+

+

+class AverageTimeReturn:

+    def __init__(self, function, total_time=0, n=0):

+        self.function = function

+        self.total_time = total_time

+        self.n = n

+

+    def __call__(self, *args, **kwargs):

+        t_start = time.time()

+        result = self.function(*args, **kwargs)

+        self.total_time += time.time() - t_start

+        self.n += 1

+        return result, self.total_time / self.n

+

+    def __getstate__(self):

+        return (self.function, self.total_time, self.n)

+

+    def __setstate__(self, state):

+        self.__init__(*state)

 def named_product(**items):

new file mode 100644

@@ -0,0 +1,19 @@

+# -*- coding: utf-8 -*-

+from contextlib import contextmanager

+from itertools import product

+

+

+def named_product(**items):

+    names = items.keys()

+    vals = items.values()

+    return [dict(zip(names, res)) for res in product(*vals)]

+

+

+@contextmanager

+def restore(*learners):

+    states = [learner.__getstate__() for learner in learners]

+    try:

+        yield

+    finally:

+        for state, learner in zip(states, learners):

+            learner.__setstate__(state)