minor tweeks

pyproject.toml

@@ -5,7 +5,7 @@ description = ""
 authors = ["kjetil midtlie <[email protected]>"]
 readme = "README.md"
 packages = [{include = "ts_language_collection", from = "src"}]
-include = ["src/**/*.so","src/**/*.dll","src/**/*pickle"]
+include = ["src/**/*.so","src/**/*.dll","src/**/*.pickle"]
 
 [tool.poetry.dependencies]
 python = "^3.10"

src/ts_language_collection/langs.py

@@ -10,14 +10,14 @@
     lang_names = pickle.load(infile)
 
 
-def get_language(language):
+def get_language(language:str) -> Language:
     assert language in lang_names, "Not a valid language option!"
     return Language(str(common.lib_filename(HERE)), language)
 
 
-def get_parser(language):
-    language = get_language(language)
+def get_parser(language:str) -> Parser:
+    lang_obj = get_language(language)
     parser = Parser()
-    parser.set_language(language)
+    parser.set_language(lang_obj)
     return parser