update

Author: justusschock

.github/workflows/ci-integrations.yaml

@@ -78,7 +78,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        lightning-source: ["pypi", "github"]
+        lightning-source: ["github"] # ["pypi", "github"] # Add once new lightning release was done
 
     timeout-minutes: 30
     env:

tests/integrations/lightning_logger_cases.py

@@ -4,7 +4,6 @@
 import uuid
 import warnings
 from contextlib import redirect_stdout
-from datetime import datetime, timezone
 from io import StringIO
 from pathlib import Path
 from time import sleep
@@ -24,12 +23,6 @@ def _unique_name(prefix: str) -> str:
     return f"{prefix}-{uuid.uuid4().hex}"
 
 
-# TODO: Remove after next lightning release
-def _patched_version() -> str:
-    timestamp = datetime.now(timezone.utc).isoformat(timespec="milliseconds")
-    return timestamp.replace(":", "-").replace("+00:00", "Z")
-
-
 def _project_and_stream_ids(logger: Any) -> tuple[str, str]:
     experiment = logger.experiment
     return experiment._teamspace.id, experiment._metrics_store.id
@@ -148,10 +141,6 @@ def run_end_to_end_smoke(logger_cls: type, *, name_prefix: str, tmpdir: Any) ->
         save_logs=False,
         checkpoint_name=checkpoint_name,
     )
-    if hasattr(logger, "_version"):
-        logger._version = _patched_version()
-    if logger.__class__.__module__ == "lightning.pytorch.loggers.litlogger":
-        logger._name = f"{run_id}-{_patched_version()}"
 
     class LitAutoEncoder(LightningModule):
         def __init__(self) -> None: