From 290f91cd5ec7f092c3fb327d62aa2fcbeae6ad7c Mon Sep 17 00:00:00 2001
From: 24seconds <34280965+24seconds@users.noreply.github.com>
Date: Sun, 28 Nov 2021 18:25:44 +0900
Subject: [PATCH] Use Either type for OptionalArg in CastArgs

Co-authored-by: Jeong YunWon <69878+youknowone@users.noreply.github.com>
---
 vm/src/builtins/memory.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vm/src/builtins/memory.rs b/vm/src/builtins/memory.rs
index cf6004e097..a34d8db3ac 100644
--- a/vm/src/builtins/memory.rs
+++ b/vm/src/builtins/memory.rs
@@ -876,7 +876,7 @@ struct CastArgs {
     #[pyarg(any)]
     format: PyStrRef,
     #[pyarg(any, optional)]
-    shape: OptionalArg<PyObjectRef>,
+    shape: OptionalArg<Either<PyTupleRef, PyListRef>>,
 }
 
 enum SubscriptNeedle {