File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -93,7 +93,7 @@ def _start_pathways_trace_from_profile_request(
9393 try :
9494 _ , result_future = _profile_state .executable .call ()
9595 result_future .result ()
96- except Exception : # pylint: disable=broad-except
96+ except Exception :
9797 _logger .exception ("Failed to start trace" )
9898 _profile_state .reset ()
9999 raise
@@ -191,7 +191,7 @@ class ProfilingConfig:
191191 repository_path : str
192192
193193 @app .post ("/profiling" )
194- async def profiling (pc : ProfilingConfig ) -> dict [str , str ]: # pylint: disable=unused-variable
194+ async def profiling (pc : ProfilingConfig ) -> dict [str , str ]:
195195 _logger .debug ("Capturing profiling data for %s ms" , pc .duration_ms )
196196 _logger .debug ("Writing profiling data to %s" , pc .repository_path )
197197 await asyncio .to_thread (jax .profiler .start_trace , pc .repository_path )
@@ -275,7 +275,7 @@ def start_trace_patch(
275275 log_dir ,
276276 create_perfetto_link : bool = False ,
277277 create_perfetto_trace : bool = False ,
278- profiler_options : jax .profiler .ProfileOptions | None = None , # pylint: disable=unused-argument
278+ profiler_options : jax .profiler .ProfileOptions | None = None ,
279279 ) -> None :
280280 _logger .debug ("jax.profile.start_trace patched with pathways' start_trace" )
281281 start_trace (
You can’t perform that action at this time.
0 commit comments