From 0607f65580aca51498ca4900a4e49138b0a41b3c Mon Sep 17 00:00:00 2001
From: Thomas D Ahle <thomas@ahle.dk>
Date: Mon, 13 Jan 2025 19:28:15 +0100
Subject: [PATCH] Missing dependency

---
 pyproject.toml          |  4 +-
 tensorgrad/testutils.py | 81 -----------------------------------------
 2 files changed, 3 insertions(+), 82 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 834bee3..196f1cd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,6 +13,7 @@ dependencies = [
    "torch",
    "nbval",
    "pytest",
+   "einops",
    "pdf2image",
    "sympy"
 ]
@@ -25,5 +26,6 @@ line-length = 110
 indent-width = 4
 
 [tool.pytest.ini_options]
-addopts = "-s"
+addopts = "-s --cov=tensorgrad --cov-report=term-missing --cov-report=xml"
 # addopts = "--nbval"
+
diff --git a/tensorgrad/testutils.py b/tensorgrad/testutils.py
index 9d28740..ff6143a 100644
--- a/tensorgrad/testutils.py
+++ b/tensorgrad/testutils.py
@@ -7,87 +7,6 @@
 from tensorgrad import Copy, Ones, Tensor, Zero, Variable
 import networkx as nx
 
-# Product(
-#     [
-#         Product(
-#             [
-#                 Product(
-#                     [
-#                         Function("max", {}, Variable("X", i, j)),
-#                         Product([Copy(i, "i")]),
-#                     ]
-#                 ),
-#                 Copy(j, "j"),
-#                 Copy(i, "i"),
-#             ]
-#         ),
-#         Function(
-#             "pow(k=-1)",
-#             {},
-#             Product(
-#                 [
-#                     Copy(j),
-#                     Copy(i),
-#                 ]
-#             ),
-#         ),
-#     ]
-# )
-
-# Product(
-#     [
-#         Product(
-#             [
-#                 Product(
-#                     [
-#                         Function(FunctionSignature(max, set(), [{"i"}]), Variable("X", i, j), {}),
-#                         Product([Copy(i, "i")]),
-#                     ]
-#                 ),
-#                 Copy(j, "j"),
-#                 Copy(i, "i"),
-#             ]
-#         ),
-#         Function(
-#             FunctionSignature(pow(k=-1), set(), [set()]),
-#             Product(
-#                 [
-#                     Copy(j),
-#                     Copy(i),
-#                 ]
-#             ),
-#             {},
-#         ),
-#     ]
-# )
-
-# Product(
-#     [
-#         Product(
-#             [
-#                 Product(
-#                     [
-#                         Function(FunctionSignature(max, set(), [{"i"}]), Variable("X", i, j), {}),
-#                         Product([Copy(i, "i")]),
-#                     ]
-#                 ),
-#                 Copy(j, "j"),
-#                 Copy(i, "i"),
-#             ]
-#         ),
-#         Function(
-#             FunctionSignature(pow(k=-1), set(), [set()]),
-#             Product(
-#                 [
-#                     Copy(j),
-#                     Copy(i),
-#                 ]
-#             ),
-#             {},
-#         ),
-#     ]
-# )
-
 
 def rand_values(variables: Iterable[Variable], shape: Dict[Symbol, int] = {}):
     values = {}