corebridge 0.2.1__tar.gz → 0.2.2__tar.gz

{corebridge-0.2.1/corebridge.egg-info → corebridge-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: corebridge
-Version: 0.2.1
+Version: 0.2.2
 Summary: Bridge for Stactics AICore
 Home-page: https://github.com/fenke/corebridge
 Author: Fenke Meijer

corebridge-0.2.2/corebridge/__init__.py

@@ -0,0 +1,157 @@
+__version__ = "0.2.1"
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/01_aicorebridge.ipynb.
+
+# %% auto 0
+__all__ = ['syslog', 'AICoreModule']
+
+# %% ../nbs/01_aicorebridge.ipynb 4
+import typing
+import logging
+import inspect
+import datetime
+import json
+import pandas as pd
+import numpy as np
+
+from fastcore.basics import patch_to, patch
+
+
+# %% ../nbs/01_aicorebridge.ipynb 5
+syslog = logging.getLogger(__name__)
+
+# %% ../nbs/01_aicorebridge.ipynb 6
+try:
+    print(f"Loading {__name__} from {__file__}")
+except:
+    pass
+
+# %% ../nbs/01_aicorebridge.ipynb 7
+class AICoreModule(): pass
+
+# %% ../nbs/01_aicorebridge.ipynb 8
+@patch
+def __init__(self:AICoreModule, 
+             processor:typing.Callable, # data processing function
+             save_dir:str, # path where the module can keep files 
+             assets_dir:str,
+             *args, **kwargs):
+    
+    self.init_time = datetime.datetime.utcnow()
+    self.save_dir   = save_dir
+    self.assets_dir = assets_dir
+    self._init_processor(processor)
+
+    self.init_args = args
+    self.init_kwargs = kwargs
+
+
+
+# %% ../nbs/01_aicorebridge.ipynb 9
+@patch
+def _init_processor(
+        self:AICoreModule, 
+        processor:typing.Callable):
+    """Initializes processor related variables on self"""
+    
+    self.processor = processor
+    self.processor_signature = inspect.signature(self.processor)
+    self.processor_params = dict(self.processor_signature.parameters)
+    self.return_param = self.processor_params.pop('return', None)
+    self.data_param, *self.call_params = list(self.processor_params.keys())
+
+
+# %% ../nbs/01_aicorebridge.ipynb 10
+@patch
+def call_processor(self:AICoreModule, calldata, **callargs):
+    return self.processor(calldata, **callargs)
+
+
+# %% ../nbs/01_aicorebridge.ipynb 11
+@patch
+def infer(self:AICoreModule, data:dict, *_, **kwargs):
+    try:
+
+        msg=[
+            f"{self.processor.__name__}({self.processor_signature})",
+            f"init_args: {self.init_args}, init_kwargs: {self.init_kwargs}",
+        ]
+
+        lastSeen = kwargs.pop('lastSeen', False)
+        recordformat = kwargs.pop('format', "records").lower()
+        reversed = kwargs.pop('reversed', False)
+        timezone = kwargs.get('timezone', 'UTC')
+        msg.append(f"lastSeen: {lastSeen}, recordformat: {recordformat}, timezone: {timezone}")
+
+        calldata = self.get_call_data(
+            data, 
+            recordformat=recordformat,
+            timezone=timezone,
+            reversed=reversed)
+        
+        msg.append(f"calldata shape: {calldata.shape}")
+
+        callargs = self.get_callargs(**kwargs)
+
+        for arg, val in callargs.items():
+            msg.append(f"{arg}: {val}")
+            
+        result = self.call_processor(calldata, **callargs)
+        msg.append(f"result shape: {result.shape}")
+
+        return {
+            'msg':msg,
+            'data': timeseries_dataframe_to_datadict(
+                result if not lastSeen else result[-1:],
+                recordformat=recordformat,
+                timezone=timezone,
+                reversed=reversed)
+        }
+    except Exception as err:
+        return {
+            'msg': f"Unexpected {err=}, {type(err)=}",
+            'data': []
+        }
+
+
+# %% ../nbs/01_aicorebridge.ipynb 12
+@patch
+def get_callargs(self:AICoreModule, **kwargs):
+    "Get arguments for the processor call"
+
+    # Remove null / None values
+    kwargs = {k:v for k,v in kwargs.items() if v is not None}
+    
+    metadata = kwargs.pop('metadata', {}) # TODO: historic metadata
+
+    return {
+        K:self.processor_signature.parameters[K].annotation(kwargs.get(K,metadata.get(K, self.init_kwargs.get(K, self.processor_signature.parameters[K].default))))
+        for K in self.call_params
+    }
+
+
+# %% ../nbs/01_aicorebridge.ipynb 13
+@patch
+def get_call_data(
+        self:AICoreModule, 
+        data:dict, 
+        recordformat='records', 
+        timezone='UTC', 
+        reversed=False):
+    
+    "Convert data to the processor signature"
+
+    df = set_time_index_zone(timeseries_dataframe_from_datadict(
+        data, ['datetimemeasure', 'time'], recordformat), timezone)
+
+    if reversed:
+        df = df[::-1]
+
+    if self.processor_params[self.data_param].annotation == pd.DataFrame:
+        return df
+    elif len(df.columns) > 1:
+        df.index = (df.index - datetime.datetime(1970,1,1, tzinfo=datetime.timezone.utc)) / datetime.timedelta(seconds=1)
+        return df.to_records(index=True)
+    else:
+        df.index = (df.index - datetime.datetime(1970,1,1, tzinfo=datetime.timezone.utc)) / datetime.timedelta(seconds=1)
+        return df.reset_index().to_numpy()
+        

{corebridge-0.2.1 → corebridge-0.2.2}/corebridge/aicorebridge.py

@@ -13,7 +13,7 @@ import pandas as pd
 import numpy as np
 
 from fastcore.basics import patch_to, patch
-import corebridge
+from core import *
 
 
 # %% ../nbs/01_aicorebridge.ipynb 5
@@ -100,7 +100,7 @@ def infer(self:AICoreModule, data:dict, *_, **kwargs):
 
         return {
             'msg':msg,
-            'data': corebridge.core.timeseries_dataframe_to_datadict(
+            'data': timeseries_dataframe_to_datadict(
                 result if not lastSeen else result[-1:],
                 recordformat=recordformat,
                 timezone=timezone,
@@ -140,7 +140,7 @@ def get_call_data(
     
     "Convert data to the processor signature"
 
-    df = corebridge.core.set_time_index_zone(corebridge.core.timeseries_dataframe_from_datadict(
+    df = set_time_index_zone(timeseries_dataframe_from_datadict(
         data, ['datetimemeasure', 'time'], recordformat), timezone)
 
     if reversed:

{corebridge-0.2.1 → corebridge-0.2.2/corebridge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: corebridge
-Version: 0.2.1
+Version: 0.2.2
 Summary: Bridge for Stactics AICore
 Home-page: https://github.com/fenke/corebridge
 Author: Fenke Meijer

{corebridge-0.2.1 → corebridge-0.2.2}/settings.ini

@@ -5,7 +5,7 @@
 ### Python library ###
 repo = corebridge
 lib_name = %(repo)s
-version = 0.2.1
+version = 0.2.2
 min_python = 3.7
 license = apache2
 black_formatting = False

corebridge-0.2.1/corebridge/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.2.1"