Raise error when instatiating cost/kernel for a grid (#565)

ott-jax · Jul 22, 2024 · 7dcc09c · 7dcc09c
1 parent 2b866d3
commit 7dcc09c
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/ott/geometry/grid.py b/src/ott/geometry/grid.py
@@ -325,6 +325,20 @@ def mask(
     """Not implemented."""
     raise NotImplementedError("Masking is not implemented for grids.")
 
+  @property
+  def cost_matrix(self) -> jnp.ndarray:
+    """Not implemented."""
+    raise NotImplementedError(
+        "Instantiating cost matrix is not implemented for grids."
+    )
+
+  @property
+  def kernel_matrix(self) -> jnp.ndarray:
+    """Not implemented."""
+    raise NotImplementedError(
+        "Instantiating kernel matrix is not implemented for grids."
+    )
+
   @classmethod
   def prepare_divergences(
       cls,