From 9f373a825c71b1e6901ed398e58dfff808afa472 Mon Sep 17 00:00:00 2001
From: Valentin BOUVAREL <valentin.bouvarel@abbeal.com>
Date: Mon, 21 Nov 2022 10:22:20 +0100
Subject: [PATCH] fix return type on gestformation

---
 gestform.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gestform.py b/gestform.py
index d70ee8f..39aa934 100644
--- a/gestform.py
+++ b/gestform.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from typing import List
+from typing import List, Union
 from random import shuffle
 
 
@@ -24,7 +24,7 @@ def randomise_with_iterator(list_from: List[int]) -> int:
         yield randomized.pop()
 
 
-def gestformation(number: int) -> str:
+def gestformation(number: int) -> Union[str, int]:
     """Transform a number to a string depends on modularity"""
     if number % GESTE == 0 and number % FORME == 0:
         return "GestForm"