checkpointer 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

checkpointer/function_body.py

@@ -1,6 +1,5 @@
 import inspect
 from types import FunctionType, CodeType
-from relib.raypipe import raypipe
 import relib.hashing as hashing
 from pathlib import Path
 from .utils import unwrap_func
@@ -18,11 +17,9 @@ def get_function_body(func):
   return '\n'.join(lines)
 
 def get_code_children(__code__):
-  return raypipe \
-    .filter(lambda const: isinstance(const, CodeType)) \
-    .flat_map(get_code_children) \
-    .do(lambda children: list(__code__.co_names) + children) \
-    .compute(__code__.co_consts)
+  consts = [const for const in __code__.co_consts if isinstance(const, CodeType)]
+  children = [child for const in consts for child in get_code_children(const)]
+  return list(__code__.co_names) + children
 
 def get_func_children(func, neighbor_funcs=[]):
   def get_candidate_func(co_name):
@@ -36,23 +33,20 @@ def get_func_children(func, neighbor_funcs=[]):
 
   code_children = get_code_children(func.__code__)
 
-  func_children = raypipe \
-    .map(get_candidate_func) \
-    .filter(clear_candidate) \
-    .compute(code_children)
+  func_children = [get_candidate_func(child) for child in code_children]
+  func_children = [child for child in func_children if clear_candidate(child)]
 
-  funcs = raypipe \
-    .flat_map(lambda child_func: \
-      get_func_children(child_func, func_children)
-    ) \
-    .do(lambda grand_children: [func] + grand_children) \
-    .sort_distinct(lambda func: func.__name__) \
-    .compute(func_children)
+  funcs = [func] + [
+    deep_child
+    for child_func in func_children
+    for deep_child in get_func_children(child_func, func_children)
+  ]
+  funcs = sorted(set(funcs), key=lambda func: func.__name__)
 
   return funcs
 
 def get_function_hash(func):
   funcs = [func] + get_func_children(func)
-  function_bodies = raypipe.map(get_function_body).compute(funcs)
+  function_bodies = list(map(get_function_body, funcs))
   function_bodies_hash = hashing.hash(function_bodies)
   return function_bodies_hash

checkpointer/utils.py

@@ -1,4 +1,4 @@
-import asyncio
+import types
 
 def unwrap_func(func):
   if hasattr(func, '__wrapped__'):
@@ -6,7 +6,7 @@ def unwrap_func(func):
   else:
     return func
 
-@asyncio.coroutine
+@types.coroutine
 def coroutine_as_generator(coroutine):
   val = yield from coroutine
   return val

{checkpointer-1.0.0.dist-info → checkpointer-1.1.0.dist-info}/METADATA

@@ -1,16 +1,12 @@
 Metadata-Version: 2.1
 Name: checkpointer
-Version: 1.0.0
-Summary: UNKNOWN
+Version: 1.1.0
 Home-page: https://github.com/Reddan/checkpointer
 Author: Hampus Hallman
 Author-email: me@hampushallman.com
 License: MIT
-Platform: UNKNOWN
 Requires-Python: ~=3.5
 License-File: LICENSE.txt
 Requires-Dist: relib
 Requires-Dist: termcolor
 
-UNKNOWN
-

{checkpointer-1.0.0.dist-info → checkpointer-1.1.0.dist-info}/RECORD

@@ -1,13 +1,13 @@
 checkpointer/__init__.py,sha256=_RYcKsZbeUf08KZ-DcXlNn4eAMWh2LN9o-KvchYVmmk,380
 checkpointer/checkpoint.py,sha256=FyL78HvAvPtgl-esiAkt-CdekT18J2Sh0SriMtX4QLc,2367
-checkpointer/function_body.py,sha256=kBsgCPeNgQzgXFToTuqQYYvAQicetXtv-UGwyd5b4Ew,1760
+checkpointer/function_body.py,sha256=aitzBFoEIo_E-aL2oe4Nr_T_Oy8ehKInGb1XYjeLzX8,1703
 checkpointer/storage.py,sha256=Ofuh0dKF5vk4_B4djt3Q6qyZhIO5f59uCNCZjMrto0U,1782
-checkpointer/utils.py,sha256=5CFQHUAlrORFpcg-TGuvAUQDeCEXvvx6yPl8mor5jh8,381
+checkpointer/utils.py,sha256=Kzh2qXxq2Lgjiqfv6KcOwXdw7sh70uCn4a_uXYwLhiM,377
 checkpointer/storages/bcolz_storage.py,sha256=Yk7FI75noe9hZBWVFIRetiFSR7tkzbryYlBmxX-lVlw,2728
 checkpointer/storages/memory_storage.py,sha256=S4SgKSApbQE-pxxKRWLNJqyZMRQwaw5-N0DOIsZM7mE,364
 checkpointer/storages/pickle_storage.py,sha256=zcnX1GG6XPHvVxi7gCab5oFxKoz5E7LZHYH74VL1hkY,1542
-checkpointer-1.0.0.dist-info/LICENSE.txt,sha256=2c7g4mni-RUemFGkk6GnoFwknh-leF04BF_J_3gp4sg,1054
-checkpointer-1.0.0.dist-info/METADATA,sha256=kmuSQOTbRa5KaFBlr48K-Fin7bRD3L7aIZiC0Lx8v6o,317
-checkpointer-1.0.0.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-checkpointer-1.0.0.dist-info/top_level.txt,sha256=uF0eyHpShnsHI3sobErnhQ8LWCT8DPViqAznKeTaZlw,13
-checkpointer-1.0.0.dist-info/RECORD,,
+checkpointer-1.1.0.dist-info/LICENSE.txt,sha256=2c7g4mni-RUemFGkk6GnoFwknh-leF04BF_J_3gp4sg,1054
+checkpointer-1.1.0.dist-info/METADATA,sha256=_isCKu9zdEI1Z-umSB6mr-X7lEVBVjlU_QvRNYolUNQ,273
+checkpointer-1.1.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+checkpointer-1.1.0.dist-info/top_level.txt,sha256=uF0eyHpShnsHI3sobErnhQ8LWCT8DPViqAznKeTaZlw,13
+checkpointer-1.1.0.dist-info/RECORD,,

{checkpointer-1.0.0.dist-info → checkpointer-1.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.0)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any