62 lines
1.1 KiB
Python
62 lines
1.1 KiB
Python
"""ProtoTorch package"""
|
|
|
|
import pkgutil
|
|
|
|
import pkg_resources
|
|
|
|
from . import (
|
|
datasets,
|
|
nn,
|
|
utils,
|
|
)
|
|
from .core import (
|
|
competitions,
|
|
components,
|
|
distances,
|
|
initializers,
|
|
losses,
|
|
pooling,
|
|
)
|
|
|
|
# Core Setup
|
|
__version__ = "0.5.0"
|
|
|
|
__all_core__ = [
|
|
"competitions",
|
|
"components",
|
|
"core",
|
|
"datasets",
|
|
"distances",
|
|
"initializers",
|
|
"losses",
|
|
"nn",
|
|
"pooling",
|
|
"utils",
|
|
]
|
|
|
|
# Plugin Loader
|
|
__path__ = pkgutil.extend_path(__path__, __name__)
|
|
|
|
|
|
def discover_plugins():
|
|
return {
|
|
entry_point.name: entry_point.load()
|
|
for entry_point in pkg_resources.iter_entry_points(
|
|
"prototorch.plugins")
|
|
}
|
|
|
|
|
|
discovered_plugins = discover_plugins()
|
|
locals().update(discovered_plugins)
|
|
|
|
# Generate combines __version__ and __all__
|
|
version_plugins = "\n".join([
|
|
"- " + name + ": v" + plugin.__version__
|
|
for name, plugin in discovered_plugins.items()
|
|
])
|
|
if version_plugins != "":
|
|
version_plugins = "\nPlugins: \n" + version_plugins
|
|
|
|
version = "core: v" + __version__ + version_plugins
|
|
__all__ = __all_core__ + list(discovered_plugins.keys())
|