cleaning up code

main.py

@@ -59,7 +59,7 @@ class PartyGen():
         self.adventurers = []
         self.adventurer_types = []
 
-    def gen_party(self):
+    def gen_party(self) -> None:
         c_id = 1
         while len(self.adventurers) < self.size:
             identifier = "adv-" + str(c_id)
@@ -75,13 +75,13 @@ class PartyGen():
                     self.adventurers.append(new_player)
                     self.adventurer_types.append(new_player.player_class)
 
-    def set_party(self, adventurers, count, level):
+    def set_party(self, adventurers: list, count: int, level: int):
         self.size = count
         self.level = level
         self.adventurers = adventurers
         self.adventurer_types = []
 
-    def get_new_party(party_size, party_level):
+    def get_new_party(party_size: int, party_level: int):
         # keep variables within expected ranges
         if party_size <= 0 or party_size > 5:
             party_size = 1
@@ -94,30 +94,28 @@ class PartyGen():
         # return the created adventurer party
         return new_party 
 
-    def get_character(self, identifer):
+    def get_character(self, identifer: str) -> Adventurer:
         for adv in self.adventurers:
             if adv.identifier == identifier:
                 return adv
 
-    def get_character_sheets(self):
+    def get_character_sheets(self) -> list:
         sheet_string = ""
         character_sheets = []
         for c in self.adventurers:
             character_sheets.append(c.vertical_sheet())
         return character_sheets
 
-    def get_json(self):
+    def get_json(self) -> str:
         party_list = []
         for c in self.adventurers:
             party_list.append(c.get_json())
         party_json = json.dumps(party_list)
         return party_json
 
-    def __str__(self):
+    def __str__(self) -> str:
         return f"{self.adventurers}"
 
-
-
 # used for local testing
 def main():
     adv_dict = Adventurer.get_subclass_dict()