Added an additional issue

Author: ajakovljevicTT

tests/TTIR/test_basic_ops.py

@@ -225,13 +225,13 @@ def module_transpose(a):
 
 
 @pytest.mark.skip(
-    "Scalar ops currently not working due to issue https://github.com/tenstorrent/tt-xla/issues/73"
+    "Scalars currently not working due to issue https://github.com/tenstorrent/tt-xla/issues/73"
 )
-def test_shape_scalar():
-    def module_shape_scalar(a):
+def test_scalar_type():
+    def module_scalar_type(a):
         return a.shape[0]
 
-    verify_module(module_shape_scalar, [(3, 3)])
+    verify_module(module_scalar_type, [(3, 3)])
 
 
 # Transpose op failing for higher ranks/dimensions.

tests/infrastructure.py

@@ -17,6 +17,7 @@ def run_on_cpu():
         yield
 
 
+# TODO(issue #80): Make this creation more explicit regarding the originating devices.
 def random_input_tensor(shape, key=42, on_device=False, dtype=jnp.float32):
     device_cpu = jax.devices("cpu")[0]
     with jax.default_device(device_cpu):
@@ -39,6 +40,7 @@ def compare_tensor_to_golden(
 ):
     ret = True
 
+    # TODO (issue #81): Remove these reshapes once the PJRT can handle scalars.
     if tensor.ndim == 0:
         tensor = tensor.reshape((1,))
     if golden.ndim == 0: