feat: better management of cli wrapper

pyproject.toml

@@ -9,6 +9,7 @@ python = ">=3.8.1,<3.10"
 pdme = "^0.9.3"
 numpy = "1.22.3"
 scipy = "1.10"
+tqdm = "^4.66.2"
 
 [tool.poetry.dev-dependencies]
 pytest = ">=6"
@@ -20,7 +21,7 @@ black = "^22.3.0"
 syrupy = "^4.0.8"
 
 [tool.poetry.scripts]
-probs = "deepdog.cli.probs:main"
+probs = "deepdog.cli.probs:wrapped_main"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]
@@ -41,6 +42,13 @@ module = [
 ]
 ignore_missing_imports = true
 
+[[tool.mypy.overrides]]
+module = [
+	"tqdm",
+	"tqdm.*"
+]
+ignore_missing_imports = true
+
 [tool.semantic_release]
 version_toml = "pyproject.toml:tool.poetry.version"
 tag_format = "{version}"