Skip to content

Commit 2343816

Browse files
authored
Refactor imports in km3netpulseextractor.py
Added conditional import for 'awkward' based on package availability for quality graphnet checks passing.
1 parent 42fe543 commit 2343816

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/graphnet/data/extractors/km3net/km3netpulseextractor.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,20 @@
11
"""Module for extracting pulse information from a KM3NeT file."""
22

3-
from typing import Any, Dict
3+
from typing import Any, Dict, TYPE_CHECKING
44
import numpy as np
55
import pandas as pd
6-
import awkward as ak
76

87
from .km3netextractor import KM3NeTExtractor
8+
from graphnet.utilities.imports import has_km3net_package
99
from graphnet.data.extractors.km3net.utilities.km3net_utilities import (
1010
create_unique_id_run_by_run,
1111
assert_no_uint_values,
1212
creating_time_zero,
1313
)
1414

15+
if has_km3net_package() or TYPE_CHECKING:
16+
import awkward as ak
17+
1518

1619
class KM3NeTPulseExtractor(KM3NeTExtractor):
1720
"""Base class for extracting pulse information from a file."""

0 commit comments

Comments
 (0)