nnlogging 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

nnlogging/utils/shell_funcs/branch_.py

@@ -1,164 +0,0 @@
-from collections.abc import Collection
-from logging import Formatter as LoggingFormatter
-from typing import Literal
-
-from rich.highlighter import Highlighter as RichHighlighter
-from rich.progress import ProgressColumn as RichProgressColumn
-from rich.theme import Theme as RichTheme
-
-from nnlogging.shell_protocol import ShellProtocol
-from nnlogging.typings import FormatTimeCallable, Omitable, Sink
-from nnlogging.utils.exception import (
-    raise_branch_exists_error,
-    raise_branch_not_found_error,
-)
-from nnlogging.utils.factory_funcs.shell_ import BranchConfig, get_branch
-from nnlogging.utils.helpers import evolve_
-
-
-def branch_configure(
-    inst: ShellProtocol,
-    /,
-    config: BranchConfig | None = None,
-    *,
-    markup: Omitable[bool] = ...,
-    highlighter: Omitable[RichHighlighter] = ...,
-    width: Omitable[int | None] = ...,
-    height: Omitable[int | None] = ...,
-    emoji: Omitable[bool] = ...,
-    color_system: Omitable[Literal["auto", "standard", "truecolor"] | None] = ...,
-    theme: Omitable[RichTheme] = ...,
-    soft_wrap: Omitable[bool] = ...,
-    force_terminal: Omitable[bool | None] = ...,
-    force_jupyter: Omitable[bool | None] = ...,
-    force_interactive: Omitable[bool | None] = ...,
-    level: Omitable[str | int] = ...,
-    show_level: Omitable[bool] = ...,
-    show_time: Omitable[bool] = ...,
-    show_path: Omitable[bool] = ...,
-    log_time_format: Omitable[str | FormatTimeCallable] = ...,
-    omit_repeated_times: Omitable[bool] = ...,
-    rich_tracebacks: Omitable[bool] = ...,
-    tracebacks_show_locals: Omitable[bool] = ...,
-    log_message_format: Omitable[str | LoggingFormatter] = ...,
-    columns: Omitable[Collection[str | RichProgressColumn]] = ...,
-    transient: Omitable[bool] = ...,
-    refresh_per_second: Omitable[float] = ...,
-    speed_estimate_period: Omitable[float] = ...,
-    default_column_markup: Omitable[bool] = ...,
-):
-    inst.branch_config = evolve_(
-        config or inst.branch_config,
-        markup=markup,
-        highlighter=highlighter,
-        width=width,
-        height=height,
-        emoji=emoji,
-        color_system=color_system,
-        theme=theme,
-        soft_wrap=soft_wrap,
-        force_terminal=force_terminal,
-        force_jupyter=force_jupyter,
-        force_interactive=force_interactive,
-        level=level,
-        show_level=show_level,
-        show_time=show_time,
-        show_path=show_path,
-        log_time_format=log_time_format,
-        omit_repeated_times=omit_repeated_times,
-        rich_tracebacks=rich_tracebacks,
-        tracebacks_show_locals=tracebacks_show_locals,
-        log_message_format=log_message_format,
-        columns=columns,
-        transient=transient,
-        refresh_per_second=refresh_per_second,
-        speed_estimate_period=speed_estimate_period,
-        default_column_markup=default_column_markup,
-    )
-
-
-def branch_add(
-    inst: ShellProtocol,
-    /,
-    name: str,
-    sink: Sink | None = None,
-    *,
-    markup: Omitable[bool] = ...,
-    highlighter: Omitable[RichHighlighter] = ...,
-    width: Omitable[int | None] = ...,
-    height: Omitable[int | None] = ...,
-    emoji: Omitable[bool] = ...,
-    color_system: Omitable[Literal["auto", "standard", "truecolor"] | None] = ...,
-    theme: Omitable[RichTheme] = ...,
-    soft_wrap: Omitable[bool] = ...,
-    force_terminal: Omitable[bool | None] = ...,
-    force_jupyter: Omitable[bool | None] = ...,
-    force_interactive: Omitable[bool | None] = ...,
-    level: Omitable[str | int] = ...,
-    show_level: Omitable[bool] = ...,
-    show_time: Omitable[bool] = ...,
-    show_path: Omitable[bool] = ...,
-    log_time_format: Omitable[str | FormatTimeCallable] = ...,
-    omit_repeated_times: Omitable[bool] = ...,
-    rich_tracebacks: Omitable[bool] = ...,
-    tracebacks_show_locals: Omitable[bool] = ...,
-    log_message_format: Omitable[str | LoggingFormatter] = ...,
-):
-    if name in inst.branches:
-        raise_branch_exists_error(
-            inst,
-            name,
-            stacklevel=6,
-        )
-    else:
-        branch = get_branch(
-            inst.branch_config,
-            sink=sink,
-            markup=markup,
-            highlighter=highlighter,
-            width=width,
-            height=height,
-            emoji=emoji,
-            color_system=color_system,
-            theme=theme,
-            soft_wrap=soft_wrap,
-            force_terminal=force_terminal,
-            force_jupyter=force_jupyter,
-            force_interactive=force_interactive,
-            level=level,
-            show_level=show_level,
-            show_time=show_time,
-            show_path=show_path,
-            log_time_format=log_time_format,
-            omit_repeated_times=omit_repeated_times,
-            rich_tracebacks=rich_tracebacks,
-            tracebacks_show_locals=tracebacks_show_locals,
-            log_message_format=log_message_format,
-        )
-        inst.branches[name] = branch
-        if inst.logger is not None:
-            # TODO: add branch-add-hint
-            inst.logger.addHandler(branch["handler"])
-
-
-def branch_remove(
-    inst: ShellProtocol,
-    /,
-    name: str,
-):
-    if name not in inst.branches:
-        raise_branch_not_found_error(
-            inst,
-            name,
-            stacklevel=6,
-        )
-    else:
-        branch = inst.branches[name]
-        branch["handler"].close()
-        branch["tasks"].clear()
-        if branch["progress"] is not None:
-            branch["progress"].stop()
-        if inst.logger is not None:
-            # TODO: add branch-remove-hint
-            inst.logger.removeHandler(branch["handler"])
-        del inst.branches[name]

nnlogging/utils/shell_funcs/run_.py

@@ -1,97 +0,0 @@
-from aim import Repo as AimRepo
-from aim.sdk.types import AimObject
-
-from nnlogging.shell_protocol import ShellProtocol
-from nnlogging.typings import Omitable
-from nnlogging.utils.factory_funcs.shell_ import RunConfig, get_aim_run
-from nnlogging.utils.helpers import evolve_
-
-
-def _run_open(
-    inst: ShellProtocol,
-    /,
-):
-    if inst.run is None:
-        inst.run = get_aim_run(inst.run_config)
-
-
-def _run_close(
-    inst: ShellProtocol,
-    /,
-):
-    if inst.run is not None:
-        inst.run.close()
-        del inst.run
-        inst.run = None
-
-
-def run_configure(
-    inst: ShellProtocol,
-    /,
-    config: RunConfig | None = None,
-    *,
-    experiment: Omitable[str | None] = ...,
-    repo: Omitable[str | AimRepo | None] = ...,
-    system_tracking_interval: Omitable[float | None] = ...,
-    capture_terminal_logs: Omitable[bool] = ...,
-    log_system_params: Omitable[bool] = ...,
-    run_hash: Omitable[str | None] = ...,
-    read_only: Omitable[bool] = ...,
-    force_resume: Omitable[bool] = ...,
-):
-    _run_close(inst)
-    inst.run_config = evolve_(
-        config or inst.run_config,
-        experiment=experiment,
-        repo=repo,
-        system_tracking_interval=system_tracking_interval,
-        capture_terminal_logs=capture_terminal_logs,
-        log_system_params=log_system_params,
-        run_hash=run_hash,
-        read_only=read_only,
-        force_resume=force_resume,
-    )
-
-
-def update_metadata(
-    inst: ShellProtocol,
-    /,
-    label: str,
-    metadata: AimObject,
-):
-    _run_open(inst)
-    assert inst.run is not None
-    inst.run[label] = metadata
-
-
-def add_tag(inst: ShellProtocol, /, tag: str):
-    _run_open(inst)
-    assert inst.run is not None
-    inst.run.add_tag(tag)
-
-
-def remove_tag(inst: ShellProtocol, /, tag: str):
-    _run_open(inst)
-    assert inst.run is not None
-    inst.run.remove_tag(tag)
-
-
-def track(
-    inst: ShellProtocol,
-    /,
-    value: object,
-    *,
-    name: str | None = None,
-    step: int | None = None,
-    epoch: int | None = None,
-    context: AimObject = None,
-):
-    _run_open(inst)
-    assert inst.run is not None
-    inst.run.track(
-        value,
-        name=name,  # pyright: ignore[reportArgumentType]
-        step=step,  # pyright: ignore[reportArgumentType]
-        epoch=epoch,  # pyright: ignore[reportArgumentType]
-        context=context,
-    )

nnlogging-0.1.2.dist-info/RECORD

@@ -1,24 +0,0 @@
-nnlogging/__init__.py,sha256=OrVcKC5BXu4t5rIwI_6208xbnl1HLtjTCFYCjJJxiUQ,211
-nnlogging/shell.py,sha256=YEYScKtaH03hGGqWkddCi1u8IrTqd38BLvCbXeEfc3s,12321
-nnlogging/shell_protocol.py,sha256=134PlZbAAscAIE4AoaWH0SVblCFydjMDTfjTr7t8DA4,1684
-nnlogging/typings/__init__.py,sha256=7ar5136odVB9vOj5MTQ6_AYuodoWOGduMSveEpVtF1Y,774
-nnlogging/typings/aliases.py,sha256=EpLqeOss5u5-zpu-tbv6g-LZ21mFyKO0PpyXmDLrCUs,2080
-nnlogging/typings/exceptions.py,sha256=rCmrwA599Uy7iOVnX12_Q8B0FMydyJX7zgOLBWxWDUM,174
-nnlogging/typings/generics.py,sha256=olwknb77voWiIydEM62aMuU5kQPyS74zoyX7Ncr9cig,680
-nnlogging/typings/protocols.py,sha256=osKmzyQ9Qzz_woxbd2eM9wDvPZ6hJZvupQ2cdkOM0H8,341
-nnlogging/utils/__init__.py,sha256=h1iOvGhiFAwO9ypaME_IQi2R0YWvFBr7tvxh1VL_d5A,1751
-nnlogging/utils/helpers.py,sha256=8KTHeyQJR36Wnm8CRG7fZsx-IX6kXU4CsSkrTEgyZ0E,785
-nnlogging/utils/exception/__init__.py,sha256=aFAbufsmio0Yndq6H7zEEoWGNA2kfO7F3ZL1p6iGYzQ,365
-nnlogging/utils/exception/branch_exists.py,sha256=kUvBWsfYIw3u7aAbEp7GWy8f2RdZL3SMtjJga7-NGVk,1122
-nnlogging/utils/exception/branch_not_found.py,sha256=K-eaA6nzlM6DWR76adPPoATD5tI49oxBmuEf2ctIeho,1133
-nnlogging/utils/exception/task_exists.py,sha256=ce2mMg2cno8fGLnTMqPZMB9IYjmU_trOf0OtaP45vxA,1240
-nnlogging/utils/exception/task_not_found.py,sha256=5Gayr_R8Z85QHYh_BlNslW_cHVOO2HljuFe_RVZQ7BU,1113
-nnlogging/utils/factory_funcs/rich_.py,sha256=RVI46Ncxk5aoMTbbQmf_mIB664IkYACXOiVOqwHLLt0,4473
-nnlogging/utils/factory_funcs/shell_.py,sha256=iv7NV0cfcuuHeq8YE70iCMmiV_2nTppQTnVJbjfGOjg,6372
-nnlogging/utils/shell_funcs/branch_.py,sha256=IfCCQ0CXEYxR0AkLbG-qmem8h7noogNk46zcUV64q4k,5615
-nnlogging/utils/shell_funcs/logger_.py,sha256=EKLHk7SoP5owbmoU5nRFEVO_xpod7wU0sT9EqxeLOgs,4667
-nnlogging/utils/shell_funcs/run_.py,sha256=jMaFUEuKq7M6X4HhTxCwDHPaDsZGw2rcVG7-SbZx4ok,2337
-nnlogging/utils/shell_funcs/task_.py,sha256=C-RHlhYt_6HXifsDalw4iMMLFmZbLjWOdxMGRXAL2zo,3206
-nnlogging-0.1.2.dist-info/METADATA,sha256=ItxmLpxl7TKpkl9Oh3YSQ60ob74HD5NMrrmuQ8RaybM,4533
-nnlogging-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-nnlogging-0.1.2.dist-info/RECORD,,