From 1b1c49808e1ededc05be6873a9195202b69b2660 Mon Sep 17 00:00:00 2001
From: Ram Rachum <ram@rachum.com>
Date: Thu, 8 Dec 2022 20:06:54 +0200
Subject: [PATCH] Fix exception causes in base_ops.py

---
 trfl/base_ops.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trfl/base_ops.py b/trfl/base_ops.py
index 1eb22fe..96e32cd 100644
--- a/trfl/base_ops.py
+++ b/trfl/base_ops.py
@@ -93,7 +93,7 @@ def wrap_rank_shape_assert(tensors_list, expected_ranks, op_name):
     error_message = ("{}: Error in rank and/or "
                      "compatibility check, {}".format(op_name, e))
     tf.logging.error(error_message)
-    raise ValueError(error_message)
+    raise ValueError(error_message) from e
 
 
 def assert_arg_bounded(value, min_value, max_value, op_name, arg_name):