From 86df4ad79c156b306a8c0402c6475c772ef0e366 Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Sat, 1 Jul 2023 21:34:38 +0100
Subject: [PATCH] Get shape to return a tuple.

---
 candle-pyo3/src/lib.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/candle-pyo3/src/lib.rs b/candle-pyo3/src/lib.rs
index 35de86c8..ab280a63 100644
--- a/candle-pyo3/src/lib.rs
+++ b/candle-pyo3/src/lib.rs
@@ -1,3 +1,4 @@
+use pyo3::types::PyTuple;
 use pyo3::exceptions::{PyTypeError, PyValueError};
 use pyo3::prelude::*;
 
@@ -27,8 +28,8 @@ impl PyTensor {
     }
 
     #[getter]
-    fn shape(&self) -> Vec<usize> {
-        self.0.dims().to_vec()
+    fn shape(&self, py: Python<'_>) -> PyObject {
+        PyTuple::new(py, self.0.dims()).to_object(py)
     }
 
     #[getter]