IntelliType is a Python library that provides enhanced type annotations with custom structure and validation capabilities. It combines the power of static type hinting with runtime type checking and validation, while ensuring full intellisense support.
- Enhanced type annotations with full intellisense support
- Runtime type checking and validation
- Integration with Pydantic for robust data validation
- Support for custom metadata in type definitions
You can install IntelliType using pip:
pip install crimson-intelli-type
Here's a simple example of how to use IntelliType:
from typing import List, Generic, TypeVar
from crimson.intelli_type import IntelliType
T = TypeVar('T')
class IntList(IntelliType[List[int]], Generic[T]):
"""Define your own type with hover capability."""
# You can use type_safe for validation if needed
valid_list: IntList[List[int]] = IntList.type_safe([1, 2, 3])
# This will raise an error due to incorrect data type
invalid_list: IntList[List[int]] = IntList.type_safe(["a", "b", "c"])IntelliType supports adding custom metadata. Use it in your own way.
class CustomTensor(IntelliType[Tuple[Tensor, Tensor], 'metadata'], Generic[T]):
"""A custom tensor type with metadata."""
print(CustomTensor.get_meta()) # Output: ('metadata',)
def forward(input_tensor: CustomTensor[Tuple[Tensor, Tensor], "(b, c, h, w), (b, 2c, h/2, w/2)"]):
print(CustomTensor.get_meta()) # Output: ('(b, c, h, w), (b, 2c, h/2, w/2)',)
...
# You can also use metadata in more complex scenarios
class Model(nn.Module):
pass
class AdvancedModel(IntelliType[nn.Module], Generic[T]):
"""General information."""
model: AdvancedModel[nn.Module, 'specific information', Model] = Model()
print(AdvancedModel.get_meta()) # Output: ('specific information', __main__.Model)Including Generic[T] in your IntelliType class definition is crucial for proper intellisense support. It allows your IDE to provide accurate type hints and autocompletion, enhancing your development experience and catching potential type errors early.
I used IntelliType to add type hints to AutoPydantic. The custom types are imported from another script. In the current script, you can easily access the type information by hovering over the variables.
^0.1.x version syntax
Please check out the example in the link below. When using IntelliType, we don't write docstrings for attributes in the function documentation. Instead, arguments are individually documented, and their docstrings are reused throughout the code. This allows you to focus the function's documentation solely on describing its purpose and behavior.
One of the biggest advantages of IntelliType is its reusability. Many arguments are often used across different functions. Traditionally, you would need to write the docstring for these arguments repeatedly.
The worst-case scenario is when you've written docstrings in multiple places and can't remember where they all are. When you modify one of your arguments, you'd need to edit all the related docstrings. However, with IntelliType, the information for your arguments is used consistently, allowing you to manage your package more cleanly and reliably.