add py.typed marker file for type checking

fepegar · Jan 29, 2022 · 8263ad4 · 8263ad4
1 parent 1c217d8
commit 8263ad4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -36,6 +36,7 @@
  'Programming Language :: Python :: 3.7',
  'Programming Language :: Python :: 3.8',
  'Programming Language :: Python :: 3.9',
+ 'Typing :: Typed',
  ],
  description=(
  'Tools for loading, augmenting and writing 3D medical images'
@@ -58,6 +59,7 @@
  include_package_data=True,
  keywords='torchio',
  name='torchio',
+ package_data={'torchio': ['py.typed']},
  packages=find_packages(include=['torchio', 'torchio.*']),
  setup_requires=[],
  test_suite='tests',

diff --git a/torchio/py.typed b/torchio/py.typed