From d632c5f190f1439061be9d0277329d8b3aeb30b1 Mon Sep 17 00:00:00 2001
From: Peter Dekkers <peter@neurallayer.com>
Date: Sat, 13 Apr 2024 21:14:18 +0200
Subject: [PATCH] Added some type info

---
 roboquant/signal.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/roboquant/signal.py b/roboquant/signal.py
index 1917016..6feb10e 100644
--- a/roboquant/signal.py
+++ b/roboquant/signal.py
@@ -32,32 +32,33 @@ class Signal:
     Signal("XYZ", 0.5, SignalType.ENTRY)
     ```
     """
+
     symbol: str
     rating: float
     type: SignalType = SignalType.ENTRY_EXIT
 
     @staticmethod
-    def buy(symbol, signal_type=SignalType.ENTRY_EXIT):
+    def buy(symbol, signal_type=SignalType.ENTRY_EXIT) -> "Signal":
         """Create a BUY signal with a rating of 1.0"""
         return Signal(symbol, 1.0, signal_type)
 
     @staticmethod
-    def sell(symbol, signal_type=SignalType.ENTRY_EXIT):
+    def sell(symbol, signal_type=SignalType.ENTRY_EXIT) -> "Signal":
         """Create a SELL signal with a rating of -1.0"""
         return Signal(symbol, -1.0, signal_type)
 
     @property
-    def is_buy(self):
+    def is_buy(self) -> bool:
         return self.rating > 0.0
 
     @property
-    def is_sell(self):
+    def is_sell(self) -> bool:
         return self.rating < 0.0
 
     @property
-    def is_entry(self):
+    def is_entry(self) -> bool:
         return SignalType.ENTRY in self.type
 
     @property
-    def is_exit(self):
+    def is_exit(self) -> bool:
         return SignalType.EXIT in self.type