Skip to content

Commit

Permalink
auto-lint code
Browse files Browse the repository at this point in the history
  • Loading branch information
ivy-dev-bot committed Sep 15, 2024
1 parent 4941cf0 commit de95d39
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 17 deletions.
2 changes: 1 addition & 1 deletion binaries.json
Original file line number Diff line number Diff line change
Expand Up @@ -149,4 +149,4 @@
}
]
}
}
}
31 changes: 18 additions & 13 deletions ivy/compiler/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,15 +2,15 @@


def clear_graph_cache():
"""Clears the graph cache which gets populated if `graph_caching` is set
to `True` in `ivy.trace_graph`, `ivy.transpile` or `ivy.unify`. Use this to
"""Clears the graph cache which gets populated if `graph_caching` is set to
`True` in `ivy.trace_graph`, `ivy.transpile` or `ivy.unify`. Use this to
reset or clear the graph cache if needed.
Examples
--------
>>> import ivy
>>> ivy.clear_graph_cache()"""

>>> ivy.clear_graph_cache()
"""
from ._compiler import clear_graph_cache as _clear_graph_cache

return _clear_graph_cache()
Expand Down Expand Up @@ -55,8 +55,8 @@ def graph_transpile(
Returns
-------
Either a transpiled Graph or a non-initialized LazyGraph."""

Either a transpiled Graph or a non-initialized LazyGraph.
"""
from ._compiler import graph_transpile as _graph_transpile

return _graph_transpile(
Expand Down Expand Up @@ -96,6 +96,7 @@ def source_to_source(
e.g. (source="torch_frontend", target="ivy") or (source="torch_frontend", target="tensorflow") etc.
Args:
----
object: The object (class/function) to be translated.
source (str, optional): The source framework. Defaults to 'torch'.
target (str, optional): The target framework. Defaults to 'tensorflow'.
Expand All @@ -106,8 +107,9 @@ def source_to_source(
the old implementation. Defaults to 'True'.
Returns:
The translated object."""

-------
The translated object.
"""
from ._compiler import source_to_source as _source_to_source

return _source_to_source(
Expand Down Expand Up @@ -138,7 +140,8 @@ def trace_graph(
params_v=None,
v=None
):
"""Takes `fn` and traces it into a more efficient composition of backend operations.
"""Takes `fn` and traces it into a more efficient composition of backend
operations.
Parameters
----------
Expand Down Expand Up @@ -208,8 +211,8 @@ def trace_graph(
>>> start = time.time()
>>> graph(x)
>>> print(time.time() - start)
0.0001785755157470703"""

0.0001785755157470703
"""
from ._compiler import trace_graph as _trace_graph

return _trace_graph(
Expand Down Expand Up @@ -249,6 +252,7 @@ def transpile(
e.g. (source="torch_frontend", target="ivy") or (source="torch_frontend", target="tensorflow") etc.
Args:
----
object: The object (class/function) to be translated.
source (str, optional): The source framework. Defaults to 'torch'.
target (str, optional): The target framework. Defaults to 'tensorflow'.
Expand All @@ -259,8 +263,9 @@ def transpile(
the old implementation. Defaults to 'True'.
Returns:
The translated object."""

-------
The translated object.
"""
from ._compiler import transpile as _transpile

return _transpile(
Expand Down
10 changes: 7 additions & 3 deletions ivy/functional/frontends/torch/serialization/serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,9 @@ def load(
mmap: Optional[bool] = None,
**pickle_load_args: Any,
) -> Any:
raise ivy.exceptions.IvyNotImplementedException("The `torch.load` frontend has not yet been implemented.")
raise ivy.exceptions.IvyNotImplementedException(
"The `torch.load` frontend has not yet been implemented."
)


def save(
Expand All @@ -23,6 +25,8 @@ def save(
pickle_module: Any = None,
pickle_protocol: int = 2,
_use_new_zipfile_serialization: bool = True,
_disable_byteorder_record: bool = False
_disable_byteorder_record: bool = False,
) -> None:
raise ivy.exceptions.IvyNotImplementedException("The `torch.save` frontend has not yet been implemented.")
raise ivy.exceptions.IvyNotImplementedException(
"The `torch.save` frontend has not yet been implemented."
)

0 comments on commit de95d39

Please sign in to comment.