Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
guarin committed Jan 25, 2024
1 parent 0b0b264 commit 4c3a5dd
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
5 changes: 3 additions & 2 deletions lightly/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,10 +79,11 @@


import os
from typing import Optional

import packaging.version
from packaging.version import Version as _Version
from packaging.version import InvalidVersion
from typing import Optional
from packaging.version import Version as _Version

# see if torchvision vision transformer is available
try:
Expand Down
3 changes: 2 additions & 1 deletion lightly/models/modules/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
# Copyright (c) 2021. Lightly AG and its affiliates.
# All Rights Reserved

from packaging.version import Version as _Version

from lightly import _timm_available, _timm_version, _torchvision_vit_available
from lightly.models.modules.heads import (
BarlowTwinsProjectionHead,
Expand All @@ -27,7 +29,6 @@
SwaVPrototypes,
)
from lightly.models.modules.nn_memory_bank import NNMemoryBankModule
from packaging.version import Version as _Version

if _torchvision_vit_available:
# Requires torchvision >=0.12
Expand Down

0 comments on commit 4c3a5dd

Please sign in to comment.