elastic-kernel 0.0.4__tar.gz → 0.0.6__tar.gz

{elastic_kernel-0.0.4/elastic_kernel.egg-info → elastic_kernel-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: elastic-kernel
-Version: 0.0.4
+Version: 0.0.6
 Summary: Elastic kernel for Jupyter with automatic checkpoint and dependency tracking
 Author-email: ryutarom <m2002r1028@icloud.com>
 License:                                  Apache License

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_kernel/kernel.py

@@ -6,7 +6,7 @@ from logging.handlers import RotatingFileHandler
 
 from ipykernel.ipkernel import IPythonKernel
 
-from elastic_notebook.elastic_notebook import ElasticNotebook
+from elastic_notebook import ElasticNotebook
 
 
 class JSTFormatter(logging.Formatter):
@@ -146,12 +146,6 @@ class ElasticKernel(IPythonKernel):
             "%Y-%m-%d %H:%M:%S.%f",
         )
 
-        # コンソールハンドラー
-        console_handler = logging.StreamHandler()
-        console_handler.setLevel(log_level)
-        console_handler.setFormatter(formatter)
-        self.logger.addHandler(console_handler)
-
         # ローテーティングファイルハンドラー
         rotating_file_handler = RotatingFileHandler(
             self.log_file_path,

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6/elastic_kernel.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: elastic-kernel
-Version: 0.0.4
+Version: 0.0.6
 Summary: Elastic kernel for Jupyter with automatic checkpoint and dependency tracking
 Author-email: ryutarom <m2002r1028@icloud.com>
 License:                                  Apache License

elastic_kernel-0.0.6/elastic_notebook/__init__.py

@@ -0,0 +1 @@
+from .elastic_notebook import ElasticNotebook  # noqa

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/common/profile_migration_speed.py

@@ -58,12 +58,5 @@ def profile_migration_speed(dirname: str, alpha=1) -> float:
     os.system("rm -rf {}".format(testing_dir))
 
     migration_speed_bps = total_bytes / (total_read_time + total_write_time * alpha)
-    print(
-        "migration speed (bps) (total_bytes / (total_read_time + total_write_time * alpha)):",
-        migration_speed_bps,
-    )
-    print("|- total bytes:", total_bytes)
-    print("|- total read time:", total_read_time)
-    print("|- total write time:", total_write_time)
-    print("|- alpha:", alpha)
+
     return migration_speed_bps

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/io/migrate.py

@@ -78,7 +78,6 @@ def migrate(
     )
 
     if filename:
-        print("Checkpoint saved to:", filename)
         write_path = filename
     else:
         write_path = FILENAME
@@ -90,9 +89,3 @@ def migrate(
             for vs in vs_list:
                 obj_list.append(shell.user_ns[vs.name])
             dill.dump(obj_list, output_file)
-    # Write the JSON file to the specified location. Uses the default location if a file path isn't specified.
-    # if filename:
-    #    print("Checkpoint saved to:", filename)
-    #    adapter.write_all(Path(filename), metadata)
-    # else:
-    # adapter.write_all(Path(FILENAME), metadata)

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/io/recover.py

@@ -39,13 +39,10 @@ def resume(filename):
 
                 return graph, fingerprint_dict, udfs, recomputation_ces, overlapping_vss
 
-            except _pickle.UnpicklingError as e:
-                print(f"Warning: Checkpoint file is corrupted: {e}")
+            except _pickle.UnpicklingError:
                 return DependencyGraph(), {}, set(), {}, []
-            except EOFError as e:
-                print(f"Warning: Checkpoint file is incomplete: {e}")
+            except EOFError:
                 return DependencyGraph(), {}, set(), {}, []
 
-    except Exception as e:
-        print(f"Error loading checkpoint: {e}")
+    except Exception:
         return DependencyGraph(), {}, set(), {}, []

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/mutation/fingerprint.py

@@ -160,8 +160,7 @@ def compare_fingerprint(
                             changed = True
                         uncomparable = True
                         fingerprint_list[2] = UncomparableObj()
-        except Exception as e:
-            print(e)
+        except Exception:
             # Variable is uncomparable
             if id_set.intersection(input_variables_id_graph_union):
                 changed = True

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/notebook/checkpoint.py

@@ -42,20 +42,15 @@ def checkpoint(
 
     # Retrieve active VSs from the graph. Active VSs are correspond to the latest instances/versions of each variable.
     active_vss = set()
-    print("---------------------------")
-    print("all variables:")
-    # print(graph.variable_snapshots)
     for vs_list in graph.variable_snapshots.values():
         if not vs_list[-1].deleted:
-            print(f"name: {vs_list[-1].name}")
             active_vss.add(vs_list[-1])
 
     # Profile the size of each variable defined in the current session.
     for active_vs in active_vss:
         # 変数がfingerprint_dictに存在するかチェック
         if active_vs.name not in fingerprint_dict:
-            print(f"Warning: Variable '{active_vs.name}' not found in fingerprint_dict")
-            continue
+            pass
 
         attr_str = getattr(shell.user_ns[active_vs.name], "__module__", None)
         # Object is unserializable
@@ -132,27 +127,11 @@ def checkpoint(
         write_log_location, notebook_name, optimizer_name
     )
     opt_end = time.time()
-    print("---------------------------")
-    print("variables to migrate:")
-    for vs in vss_to_migrate:
-        print(f"name: {vs.name}, size: {vs.size}")
 
     difference_start = time.time()
     vss_to_recompute = active_vss - vss_to_migrate
     difference_end = time.time()
 
-    print("---------------------------")
-    print("variables to recompute:")
-    for vs in vss_to_recompute:
-        print(f"name: {vs.name}, size: {vs.size}")
-    print([vs.name for vs in vss_to_recompute])
-
-    print("---------------------------")
-    print("cells to recompute:")
-    for ce in ces_to_recompute:
-        print(f"cell num: {ce.cell_num}, cell runtime: {ce.cell_runtime}")
-    print(sorted([ce.cell_num + 1 for ce in ces_to_recompute]))
-
     optimize_end = time.time()
     if write_log_location:
         with open(

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/core/notebook/restore_notebook.py

@@ -34,9 +34,6 @@ def restore_notebook(
     recompute_start = time.time()
     for ce in graph.cell_executions:
         if ce in ces_to_recompute:
-            # Rerun cell code; suppress stdout when rerunning.
-            print("Rerunning cell", ce.cell_num + 1)
-
             # 出力を完全に抑制するための設定
             ipython = get_ipython()
 
@@ -68,7 +65,6 @@ def restore_notebook(
 
         # Define output variables in the CE.
         for pair in variables[ce.cell_num]:
-            print("Declaring variable", pair[0].name)
             shell.user_ns[pair[0].name] = pair[1]
 
     recompute_end = time.time()

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/elastic_notebook/elastic_notebook.py

@@ -69,9 +69,6 @@ class ElasticNotebook:
         # Dict for recording overhead of profiling operations.
         self.profile_dict = {"idgraph": 0, "representation": 0}
 
-        # デバッグフラグ
-        self.debug = False
-
         # マイグレーションと再計算の変数リスト
         self._vss_to_migrate = []
         self._vss_to_recompute = []
@@ -95,13 +92,7 @@ class ElasticNotebook:
         """文字列表現を定義"""
         return f"マイグレーション対象: {self.vss_to_migrate}\n再計算対象: {self.vss_to_recompute}"
 
-    def set_debug(self, debug):
-        self.debug = debug
-
     def record_event(self, cell):
-        if self.debug:
-            print("Recording event...")
-
         pre_execution = set(self.shell.user_ns.keys())
 
         # Create id trees for output variables
@@ -215,11 +206,9 @@ class ElasticNotebook:
             if float(migration_speed) > 0:
                 self.migration_speed_bps = float(migration_speed)
                 self.manual_migration_speed = True
-            else:
-                pass
-                # print("Migration speed is not positive.")
         except ValueError:
-            print("Migration speed is not a number.")
+            pass
+
         self.selector.migration_speed_bps = self.migration_speed_bps
 
     def set_optimizer(self, optimizer):
@@ -247,9 +236,6 @@ class ElasticNotebook:
 
     def checkpoint(self, filename):
         """チェックポイントを作成"""
-        if self.debug:
-            print("Checkpointing...")
-
         # Write overhead metrics to file (for experiments).
         if self.write_log_location:
             with open(
@@ -312,10 +298,6 @@ class ElasticNotebook:
         )
 
     def load_checkpoint(self, filename):
-        if self.debug:
-            print("Loading checkpoint...")
-
-        # start_time = time.time()
         (
             self.dependency_graph,
             variables,
@@ -333,4 +315,3 @@ class ElasticNotebook:
             self.notebook_name,
             self.optimizer_name,
         )
-        # print("Checkpoint load time:", time.time() - start_time)

{elastic_kernel-0.0.4 → elastic_kernel-0.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "elastic-kernel"
-version = "0.0.4"
+version = "0.0.6"
 description = "Elastic kernel for Jupyter with automatic checkpoint and dependency tracking"
 authors = [
     { name = "ryutarom", email = "m2002r1028@icloud.com" }