add type hinting

2020-09-11 03:00:39 +02:00
parent 524aedabd5
commit d580e9f91a
1 changed files with 7 additions and 3 deletions
--- a/strat/main.py
+++ b/strat/main.py
@@ -1,10 +1,14 @@
 # Y8O353, E0HJQL

-import numpy as np
+def avg(ls: list) -> int:
+    if len(ls) != 0:
+        return sum(ls) / len(ls)
+    else:
+        return 0


-def decision(h1, h2):
-    if 1 in h2[:4]:
+def decision(h1: list, h2: list) -> int:
+    if avg(h2[:4]) >= 0.05:
        return 1
    else:
        return 0