pfund 0.0.1.dev12__tar.gz → 0.0.1.dev13__tar.gz

{pfund-0.0.1.dev12 → pfund-0.0.1.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pfund
-Version: 0.0.1.dev12
+Version: 0.0.1.dev13
 Summary: A Complete Algo-Trading Framework for Machine Learning, enabling trading across TradFi, CeFi and DeFi. Supports Vectorized and Event-Driven Backtesting, Paper and Live Trading
 Home-page: https://pfund.ai
 License: Apache-2.0
@@ -21,7 +21,7 @@ Requires-Dist: gitpython (>=3.1.43,<4.0.0)
 Requires-Dist: mlflow (>=2.11.3,<3.0.0) ; extra == "ml"
 Requires-Dist: orjson (>=3.9.14,<4.0.0) ; extra == "data"
 Requires-Dist: papermill (>=2.5.0,<3.0.0) ; extra == "analytics"
-Requires-Dist: pfeed[boost,data,df] (>=0.0.1.dev10,<0.0.2) ; extra == "data"
+Requires-Dist: pfeed[boost,data,df] (>=0.0.1.dev11,<0.0.2) ; extra == "data"
 Requires-Dist: pfolio[bayesian,data,portfolio,temporary] (>=0.0.1.dev4,<0.0.2) ; extra == "analytics"
 Requires-Dist: platformdirs (>=4.2.0,<5.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)

pfund-0.0.1.dev13/pfund/__init__.py

@@ -0,0 +1,70 @@
+import sys
+from importlib.metadata import version
+
+from pfund.config_handler import configure
+from pfund.utils.aliases import ALIASES
+from pfund.const.paths import PROJ_PATH
+# add python path so that for files like "ibapi" (official python code from IB) can find their modules
+sys.path.append(f'{PROJ_PATH}/externals')
+
+
+# NOTE: dynamically import modules to avoid click cli latency (reduced from ~4s to ~0.2s)
+def __getattr__(name):
+    """
+    Dynamically import and return modules and classes based on their name.
+    
+    Supports dynamic loading of data sources and feed classes to minimize
+    initial load time.
+    """
+    import importlib
+    if 'Engine' in name:
+        Engine = getattr(importlib.import_module('pfund.engines'), name)
+        globals()[name] = Engine
+        return Engine
+    elif 'Strategy' in name:
+        Strategy = getattr(importlib.import_module('pfund.strategies'), name)
+        globals()[name] = Strategy
+        return Strategy
+    elif 'Model' in name or 'Feature' in name:
+        Model = getattr(importlib.import_module('pfund.models'), name)
+        globals()[name] = Model
+        return Model
+    elif 'Indicator' in name:
+        Indicator = getattr(importlib.import_module('pfund.indicators'), name)
+        globals()[name] = Indicator
+        return Indicator
+
+
+# NOTE: dummy classes/modules for type hinting
+# e.g. import pfund as pf, when you type "pf.", 
+# you will still see the following suggestions even they are dynamically imported:
+BacktestEngine: ...
+TradeEngine: ...
+TrainEngine: ...
+TestEngine: ...
+Strategy: ...
+Model: ...
+PyTorchModel: ...
+SKLearnModel: ...
+Feature: ...
+TAIndicator: ...
+TALibIndicator: ...
+
+
+__version__ = version('pfund')
+__all__ = (
+    '__version__',
+    'configure', 
+    'ALIASES',
+    'BacktestEngine', 
+    'TradeEngine',
+    'TrainEngine', 
+    'TestEngine',
+    'Strategy', 
+    'Model', 
+    'PyTorchModel', 
+    'SKLearnModel',
+    'Feature', 
+    'TAIndicator', 
+    'TALibIndicator',
+)

pfund-0.0.1.dev13/pfund/main.py

@@ -0,0 +1,18 @@
+import atexit
+
+from pfund.cli import pfund_group
+
+
+def exit_cli():
+    """Application Exitpoint."""
+    print("Cleanup actions here...")
+
+
+def run_cli() -> None:
+    """Application Entrypoint."""
+    # atexit.register(exit_cli)
+    pfund_group(obj={})
+
+
+if __name__ == '__main__':
+    run_cli()

{pfund-0.0.1.dev12 → pfund-0.0.1.dev13}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pfund"
-version = "0.0.1.dev12"
+version = "0.0.1.dev13"
 description = "A Complete Algo-Trading Framework for Machine Learning, enabling trading across TradFi, CeFi and DeFi. Supports Vectorized and Event-Driven Backtesting, Paper and Live Trading"
 license = "Apache-2.0"
 authors = ["Stephen Yau <softwareentrepreneer+pfund@gmail.com>"]
@@ -23,7 +23,7 @@ requests = "^2.31.0"
 websocket-client = "^1.7.0"
 python-telegram-bot = "^20.7"
 gitpython = "^3.1.43"
-pfeed = { version = "^0.0.1.dev10", optional = true, extras = ["df", "data", "boost"] }
+pfeed = { version = "^0.0.1.dev11", optional = true, extras = ["df", "data", "boost"] }
 orjson = { version = "^3.9.14", optional = true }
 pyzmq = { version = "^25.1.2", optional = true }
 ta = { version = "^0.11.0", optional = true }
@@ -50,6 +50,10 @@ optional = true
 pfolio = { path = "../pfolio", develop = true, extras = ["bayesian", "data", "portfolio", "temporary"] }
 pybit = "^5.6.2"
 ta-lib = "^0.4.28"
+commitizen = "^3.24.0"
+mypy = "^1.9.0"
+ruff = "^0.3.5"
+grayskull = "^2.5.3"
 
 [tool.poetry.group.test.dependencies]
 pytest = "^8.0.0"
@@ -58,11 +62,8 @@ pytest-cov = "^5.0.0"
 pytest-xdist = "^3.5.0"
 tox = "^4.14.2"
 faker = "^24.4.0"
-pre-commit = "^3.6.1"
 bandit = "^1.7.7"
-grayskull = "^2.5.3"
-mypy = "^1.9.0"
-ruff = "^0.3.5"
+pre-commit = "^3.6.1"
 
 [tool.poetry.group.doc.dependencies]
 jupyter-book = "^1.0.0"

pfund-0.0.1.dev12/pfund/__init__.py

@@ -1,32 +0,0 @@
-import sys
-from importlib.metadata import version
-
-from pfund.const.paths import PROJ_PATH
-# add python path so that for files like "ibapi" (official python code from IB) can find their modules
-sys.path.append(f'{PROJ_PATH}/externals')
-from pfund.config_handler import configure
-from pfund.engines import BacktestEngine, TrainEngine, TestEngine, TradeEngine
-from pfund.strategies import Strategy
-from pfund.models import Feature, Model
-from pfund.utils.aliases import ALIASES
-try:
-    from pfund.models import PyTorchModel
-except ImportError:
-    pass
-try:
-    from pfund.models import SKLearnModel
-except ImportError:
-    pass
-from pfund.indicators import TAIndicator, TALibIndicator
-
-
-__version__ = version('pfund')
-
-
-__all__ = (
-    '__version__',
-    'configure', 'ALIASES',
-    'BacktestEngine', 'TrainEngine', 'TestEngine', 'TradeEngine',
-    'Strategy', 'Model', 'PyTorchModel', 'SKLearnModel',
-    'Feature', 'TAIndicator', 'TALibIndicator',
-)

pfund-0.0.1.dev12/pfund/main.py

@@ -1,10 +0,0 @@
-from pfund.cli import pfund_group
-
-
-def run_cli() -> None:
-    """Application Entrypoint."""
-    pfund_group(obj={})
-
-
-if __name__ == '__main__':
-    run_cli()