Skip to content

Commit

Permalink
flake8 & black & isort reformatting
Browse files Browse the repository at this point in the history
  • Loading branch information
yehudaorel committed Jul 8, 2024
1 parent 0cbb079 commit 7d95373
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
2 changes: 1 addition & 1 deletion python/ctranslate2/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@
)
from ctranslate2.extensions import register_extensions
from ctranslate2.logging import get_log_level, set_log_level
from ctranslate2.profiling import init_profiler, dump_profiler
from ctranslate2.profiling import dump_profiler, init_profiler

register_extensions()
del register_extensions
Expand Down
6 changes: 4 additions & 2 deletions python/ctranslate2/profiling.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,9 @@
from ctranslate2 import _ext, Device
import sys

def init_profiler(device = Device.cpu, num_threads = 1):
from ctranslate2 import Device, _ext


def init_profiler(device=Device.cpu, num_threads=1):
_ext.init_profiling(device, num_threads)


Expand Down

0 comments on commit 7d95373

Please sign in to comment.