Update tracetools_read imports
This commit is contained in:
parent
45bc931937
commit
a7951796d4
5 changed files with 10 additions and 9 deletions
|
@ -16,7 +16,8 @@
|
||||||
|
|
||||||
from pickle import Pickler
|
from pickle import Pickler
|
||||||
|
|
||||||
from tracetools_read import utils
|
from tracetools_read import _get_trace_ctf_events
|
||||||
|
from tracetools_read import event_to_dict
|
||||||
|
|
||||||
|
|
||||||
def ctf_to_pickle(trace_directory: str, target: Pickler) -> int:
|
def ctf_to_pickle(trace_directory: str, target: Pickler) -> int:
|
||||||
|
@ -27,7 +28,7 @@ def ctf_to_pickle(trace_directory: str, target: Pickler) -> int:
|
||||||
:param target: the target pickle file to write to
|
:param target: the target pickle file to write to
|
||||||
:return: the number of events written
|
:return: the number of events written
|
||||||
"""
|
"""
|
||||||
ctf_events = utils._get_trace_ctf_events(trace_directory)
|
ctf_events = _get_trace_ctf_events(trace_directory)
|
||||||
|
|
||||||
count = 0
|
count = 0
|
||||||
count_written = 0
|
count_written = 0
|
||||||
|
@ -35,7 +36,7 @@ def ctf_to_pickle(trace_directory: str, target: Pickler) -> int:
|
||||||
for event in ctf_events:
|
for event in ctf_events:
|
||||||
count += 1
|
count += 1
|
||||||
|
|
||||||
pod = utils.event_to_dict(event)
|
pod = event_to_dict(event)
|
||||||
target.dump(pod)
|
target.dump(pod)
|
||||||
count_written += 1
|
count_written += 1
|
||||||
|
|
||||||
|
|
|
@ -21,9 +21,9 @@ from typing import List
|
||||||
from typing import Set
|
from typing import Set
|
||||||
from typing import Type
|
from typing import Type
|
||||||
|
|
||||||
from tracetools_read.utils import DictEvent
|
from tracetools_read import DictEvent
|
||||||
from tracetools_read.utils import get_event_name
|
from tracetools_read import get_event_name
|
||||||
from tracetools_read.utils import get_field
|
from tracetools_read import get_field
|
||||||
|
|
||||||
|
|
||||||
class EventMetadata():
|
class EventMetadata():
|
||||||
|
|
|
@ -16,7 +16,7 @@
|
||||||
|
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
from tracetools_read.utils import get_field
|
from tracetools_read import get_field
|
||||||
|
|
||||||
from . import EventHandler
|
from . import EventHandler
|
||||||
from . import EventMetadata
|
from . import EventMetadata
|
||||||
|
|
|
@ -21,7 +21,7 @@ from typing import Tuple
|
||||||
from typing import Type
|
from typing import Type
|
||||||
from typing import Union
|
from typing import Union
|
||||||
|
|
||||||
from tracetools_read.utils import get_field
|
from tracetools_read import get_field
|
||||||
|
|
||||||
from . import EventHandler
|
from . import EventHandler
|
||||||
from . import EventMetadata
|
from . import EventMetadata
|
||||||
|
|
|
@ -16,7 +16,7 @@
|
||||||
|
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
from tracetools_read.utils import get_field
|
from tracetools_read import get_field
|
||||||
|
|
||||||
from . import EventHandler
|
from . import EventHandler
|
||||||
from . import EventMetadata
|
from . import EventMetadata
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue