commit 4b26857cbecdd70127288d9c495843661300dfc6
author Luke Chang <lukechang@google.com> Tue Aug 30 12:19:42 2022 +0800
committer Luke Chang <lukechang@google.com> Tue Aug 30 12:21:56 2022 +0800
tree d53358e426733ac5abfb2c885be55d23a8446d05
parent 437f4431c5225df1e2debd6309ae8096f63c6eda

Fix typo for TraceUriResolver example.

Signed-off-by: Luke Chang <lukechang@google.com>
Change-Id: I7bcfb9c251d134900c4bf08de079355923747c8b

Bug: 244259925
Test: build pass
Change-Id: I9c7259c2036aecc3fa6f3dd40c927bc492e9e00d

python/perfetto/batch_trace_processor/api.py

diff --git a/python/perfetto/batch_trace_processor/api.py b/python/perfetto/batch_trace_processor/api.py
index 919cc23..f6f353a 100644
--- a/python/perfetto/batch_trace_processor/api.py
+++ b/python/perfetto/batch_trace_processor/api.py
@@ -236,13 +236,13 @@
       value in the dataframe).
 
       For example:
-        class CustomResolver(TraceResolver):
+        class CustomResolver(TraceUriResolver):
           def resolve(self):
-            return [TraceResolver.Result(trace='/tmp/path',
-                                        metadata={
-                                          'path': '/tmp/path'
-                                          'foo': 'bar'
-                                        })]
+            return [TraceUriResolver.Result(trace='/tmp/path',
+                                            metadata={
+                                              'path': '/tmp/path'
+                                              'foo': 'bar'
+                                            })]
 
         with BatchTraceProcessor(CustomResolver()) as btp:
           df = btp.query_and_flatten('select count(1) as cnt from slice')

python/perfetto/trace_uri_resolver/resolver.py

diff --git a/python/perfetto/trace_uri_resolver/resolver.py b/python/perfetto/trace_uri_resolver/resolver.py
index 7344dd4..bcdbf93 100644
--- a/python/perfetto/trace_uri_resolver/resolver.py
+++ b/python/perfetto/trace_uri_resolver/resolver.py
@@ -54,7 +54,7 @@
   up traces using URI strings.
 
   For example:
-    class CustomTraceResolver(TraceResolver):
+    class CustomTraceResolver(TraceUriResolver):
       PREFIX = 'custom'
 
       def __init__(self, build_branch: List[str] = None, id: str = None):
@@ -66,7 +66,7 @@
         traces = self.db.lookup(
           id=self.id, build_branch=self.build_branch)['path']
         return [
-          TraceResolver.Result(
+          TraceUriResolver.Result(
             trace=t['path'],
             args={'iteration': t['iteration'], 'device': t['device']}
           )