{"id":23350783,"url":"https://github.com/twsl/lit-mlflow","last_synced_at":"2025-10-08T17:12:22.484Z","repository":{"id":269109717,"uuid":"790336215","full_name":"twsl/lit-mlflow","owner":"twsl","description":"An improved Lightning mlflow logger to replace mlflow.autolog()","archived":false,"fork":false,"pushed_at":"2025-09-09T02:37:41.000Z","size":1467,"stargazers_count":0,"open_issues_count":10,"forks_count":0,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-09-09T04:58:41.933Z","etag":null,"topics":["databricks","lightning","mlflow","pytorch","pytorch-lightning"],"latest_commit_sha":null,"homepage":"https://twsl.github.io/lit-mlflow/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/twsl.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":".github/CONTRIBUTING.md","funding":null,"license":"LICENSE","code_of_conduct":"CODE_OF_CONDUCT.md","threat_model":null,"audit":null,"citation":null,"codeowners":".github/CODEOWNERS","security":"SECURITY.md","support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null,"notice":null,"maintainers":null,"copyright":null,"agents":null,"dco":null,"cla":null}},"created_at":"2024-04-22T17:37:22.000Z","updated_at":"2025-09-07T09:10:51.000Z","dependencies_parsed_at":"2024-12-21T01:27:29.527Z","dependency_job_id":"a7563b4b-8f43-438c-88c4-e187b08a6d9a","html_url":"https://github.com/twsl/lit-mlflow","commit_stats":null,"previous_names":["twsl/lit-mlflow"],"tags_count":1,"template":false,"template_full_name":null,"purl":"pkg:github/twsl/lit-mlflow","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/twsl%2Flit-mlflow","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/twsl%2Flit-mlflow/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/twsl%2Flit-mlflow/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/twsl%2Flit-mlflow/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/twsl","download_url":"https://codeload.github.com/twsl/lit-mlflow/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/twsl%2Flit-mlflow/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":278981518,"owners_count":26079640,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","status":"online","status_checked_at":"2025-10-08T02:00:06.501Z","response_time":56,"last_error":null,"robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":true,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["databricks","lightning","mlflow","pytorch","pytorch-lightning"],"created_at":"2024-12-21T08:18:08.093Z","updated_at":"2025-10-08T17:12:22.471Z","avatar_url":"https://github.com/twsl.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# lit-mlflow\n\n\n[![Build](https://github.com/twsl/lit-mlflow/actions/workflows/build.yaml/badge.svg)](https://github.com/twsl/lit-mlflow/actions/workflows/build.yaml)\n[![Documentation](https://github.com/twsl/lit-mlflow/actions/workflows/docs.yaml/badge.svg)](https://github.com/twsl/lit-mlflow/actions/workflows/docs.yaml)\n![GitHub Release](https://img.shields.io/github/v/release/twsl/lit-mlflow?include_prereleases)\n[![PyPI - Package Version](https://img.shields.io/pypi/v/lit-mlflow?logo=pypi\u0026style=flat\u0026color=orange)](https://pypi.org/project/lit-mlflow/)\n[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/lit-mlflow?logo=pypi\u0026style=flat\u0026color=blue)](https://pypi.org/project/lit-mlflow/)\n[![Dependencies Status](https://img.shields.io/badge/dependencies-up%20to%20date-brightgreen.svg)](https://github.com/twsl/lit-mlflow/pulls?utf8=%E2%9C%93\u0026q=is:pr%20author:app/dependabot)\n[![Docs with MkDocs](https://img.shields.io/badge/MkDocs-docs?style=flat\u0026logo=materialformkdocs\u0026logoColor=white\u0026color=%23526CFE)](https://squidfunk.github.io/mkdocs-material/)\n[![Poetry](https://img.shields.io/endpoint?url=https://python-poetry.org/badge/v0.json)](https://python-poetry.org/)\n[![linting: ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)\n[![pre-commit](https://img.shields.io/badge/pre--commit-enabled-brightgreen?logo=pre-commit)](.pre-commit-config.yaml)\n[![Checked with pyright](https://microsoft.github.io/pyright/img/pyright_badge.svg)](https://microsoft.github.io/pyright/)\n[![security: bandit](https://img.shields.io/badge/security-bandit-yellow.svg)](https://github.com/PyCQA/bandit)\n[![Semantic Versions](https://img.shields.io/badge/%20%20%F0%9F%93%A6%F0%9F%9A%80-semantic--versions-e10079.svg)](https://github.com/twsl/lit-mlflow/releases)\n[![Copier](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/copier-org/copier/master/img/badge/badge-grayscale-border.json)](https://github.com/copier-org/copier)\n[![License](https://img.shields.io/badge/license-MIT-blue)](LICENSE)\n\n\nAn improved Lightning mlflow logger. Works seamlessly with PyTorch Lightning on Databricks and offers more control compared to the `mlflow.pytorch.autolog` function.\n\n\n## Features\n\n- Makes `MLflow` logging work with `lightning` and [Databricks](https://www.databricks.com/)\n\n\n## Installation\n\nWith `pip`:\n```bash\npython -m pip install lit-mlflow\n```\n\nWith [`poetry`](https://python-poetry.org/):\n```bash\npoetry add lit-mlflow\n```\n\n\n## How to use it\nReplace `mlflow.autolog()` with the `MlFlowAutoCallback`:\n\n```python\nfrom lit_mlflow import MlFlowAutoCallback\nimport lightning.pytorch as pl\n\ntrainer = pl.Trainer(\n    callbacks=[\n        MlFlowAutoCallback()\n    ]\n)\n```\n\nTo support Databricks mlflow, use the `DbxMLFlowLogger` instead of the `MlFlowLogger`:\n\n```python\nfrom lit_mlflow import DbxMLFlowLogger\nimport lightning.pytorch as pl\n\ntrainer = pl.Trainer(\n    logger=[\n        DbxMLFlowLogger()\n    ]\n)\n```\n\n## Docs\n\n```bash\npoetry run mkdocs build -f ./docs/mkdocs.yml -d ./_build/\n```\n\n\n## Update template\n\n```bash\ncopier update --trust -A --vcs-ref=HEAD\n```\n\n\n## Credits\n\nThis project was generated with [![🚀 A generic python project template.](https://img.shields.io/badge/python--project--template-%F0%9F%9A%80-brightgreen)](https://github.com/twsl/python-project-template)\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ftwsl%2Flit-mlflow","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Ftwsl%2Flit-mlflow","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Ftwsl%2Flit-mlflow/lists"}