cogforge-engine 2.1.0__tar.gz → 2.1.1__tar.gz

{cogforge_engine-2.1.0 → cogforge_engine-2.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cogforge-engine
-Version: 2.1.0
+Version: 2.1.1
 Summary: A custom autograd engine and Transformer block built from scratch.
 Project-URL: Homepage, https://github.com/avikmjd2/cogforge
 Author-email: Avik Majumder <avikmjd2@gmail.com>

{cogforge_engine-2.1.0 → cogforge_engine-2.1.1}/cogforge/utils.py

@@ -1,4 +1,5 @@
 from graphviz import Digraph
+import numpy
 
 def trace_graph(root_tensor):
     """Recursively traces the tensor graph to find all nodes and edges."""
@@ -14,25 +15,43 @@ def trace_graph(root_tensor):
     build(root_tensor)
     return nodes, edges
 
-def draw_dot(root_tensor, format='svg', rankdir='LR'):
+def draw_graph(root_tensor, max_char=50,format='svg', rankdir='LR'):
     """
     Renders a computational graph starting from the given root_tensor.
     format: output format (e.g., 'png', 'svg', 'pdf')
     rankdir: 'LR' for left-to-right layout, 'TB' for top-to-bottom
     """
     dot = Digraph(format=format, graph_attr={'rankdir': rankdir})
-    
     nodes, edges = trace_graph(root_tensor)
     
+    def format_array(arr):
+        """Sanitizes and truncates array strings for safe Graphviz rendering."""
+        if arr is None:
+            return "None"
+        if isinstance(arr, numpy.ndarray):
+            s = numpy.array2string(arr, precision=4, suppress_small=True, separator=', ')
+        else:
+            s = str(arr)
+            
+        s = s.replace('\n', '')
+        
+        if len(s) > max_char:
+            s = s[:max_char-3] + "..."
+        s = s.replace('{', '\\{').replace('}', '\\}').replace('|', '\\|')
+        return s
+
     for n in nodes:
         uid = str(id(n))
         shape_str = str(n.shape)
         
+        data_str = format_array(n.data)
+        grad_str = format_array(n.grad)
+        
         if n.op:
-            label = f"{{ {n.op} | shape: {shape_str} }}"
+            label = f"{{ {n.op} | shape: {shape_str} | data: {data_str} | grad: {grad_str} }}"
             dot.node(name=uid, label=label, shape='record', style='filled', fillcolor='white')
         else:
-            label = f"{{ Data | shape: {shape_str} }}"
+            label = f"{{ Node | shape: {shape_str} | data: {data_str} | grad: {grad_str} }}"
             dot.node(name=uid, label=label, shape='record', style='filled', fillcolor='lightblue')
             
     for child, parent in edges:

{cogforge_engine-2.1.0 → cogforge_engine-2.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cogforge-engine" 
-version = "2.1.0"
+version = "2.1.1"
 authors = [
   { name="Avik Majumder", email="avikmjd2@gmail.com" },
 ]