Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
fairscale
Commits
3fbde78f
Unverified
Commit
3fbde78f
authored
Feb 28, 2023
by
Min Xu
Committed by
GitHub
Feb 28, 2023
Browse files
make a logging warning once (#1108)
- fixes #1107 Co-authored-by:
Min Xu
<
min.xu.public@gmail.com
>
parent
da0cbd7c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
fairscale/internal/version.py
fairscale/internal/version.py
+4
-1
No files found.
fairscale/internal/version.py
View file @
3fbde78f
...
@@ -11,13 +11,15 @@ import torch
...
@@ -11,13 +11,15 @@ import torch
__all__
:
List
[
str
]
=
[
"torch_version"
]
__all__
:
List
[
str
]
=
[
"torch_version"
]
_logged
=
False
def
torch_version
(
version
:
str
=
torch
.
__version__
)
->
Tuple
[
int
,
...]:
def
torch_version
(
version
:
str
=
torch
.
__version__
)
->
Tuple
[
int
,
...]:
numbering
=
re
.
search
(
r
"^(\d+).(\d+).(\d+)([^\+]*)(\+\S*)?$"
,
version
)
numbering
=
re
.
search
(
r
"^(\d+).(\d+).(\d+)([^\+]*)(\+\S*)?$"
,
version
)
if
not
numbering
:
if
not
numbering
:
return
tuple
()
return
tuple
()
# Catch torch version if run against internal pre-releases, like `1.8.0a0fb`,
# Catch torch version if run against internal pre-releases, like `1.8.0a0fb`,
if
numbering
.
group
(
4
):
global
_logged
if
numbering
.
group
(
4
)
and
not
_logged
:
# Two options here:
# Two options here:
# - either skip this version (minor number check is not relevant)
# - either skip this version (minor number check is not relevant)
# - or check that our codebase is not broken by this ongoing development.
# - or check that our codebase is not broken by this ongoing development.
...
@@ -25,5 +27,6 @@ def torch_version(version: str = torch.__version__) -> Tuple[int, ...]:
...
@@ -25,5 +27,6 @@ def torch_version(version: str = torch.__version__) -> Tuple[int, ...]:
# Assuming that we're interested in the second use-case more than the first,
# Assuming that we're interested in the second use-case more than the first,
# return the pre-release or dev numbering
# return the pre-release or dev numbering
logging
.
warning
(
f
"Pytorch pre-release version
{
version
}
- assuming intent to test it"
)
logging
.
warning
(
f
"Pytorch pre-release version
{
version
}
- assuming intent to test it"
)
_logged
=
True
return
tuple
(
int
(
numbering
.
group
(
n
))
for
n
in
range
(
1
,
4
))
return
tuple
(
int
(
numbering
.
group
(
n
))
for
n
in
range
(
1
,
4
))
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment