From 2666ef7aebc218674f2354e0e5a0080b1f904ba1 Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Fri, 22 Sep 2023 11:41:46 -0500
Subject: [PATCH] Using "UNion" instead of "|" for documenting return type to
 be python 3.9-compatible

---
 control.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/control.py b/control.py
index 18227bf..47303e2 100644
--- a/control.py
+++ b/control.py
@@ -1,3 +1,4 @@
+from typing import Union
 import torch
 
 import comfy.utils
@@ -31,7 +32,7 @@ class LatentKeyframeGroup:
             self.keyframes.append(keyframe)
         self.keyframes.sort(key=lambda k: k.batch_index)
     
-    def get_index(self, index: int) -> LatentKeyframe | None:
+    def get_index(self, index: int) -> Union[LatentKeyframe, None]:
         try:
             return self.keyframes[index]
         except IndexError:
@@ -79,7 +80,7 @@ class TimestepKeyframeGroup:
             self.keyframes.append(keyframe)
         self.keyframes.sort(key=lambda k: k.start_percent)
 
-    def get_index(self, index: int) -> TimestepKeyframe | None:
+    def get_index(self, index: int) -> Union[TimestepKeyframe, None]:
         try:
             return self.keyframes[index]
         except IndexError: