Skip to content

Commit df630e5

Browse files
committed
ruff format
1 parent 392fae0 commit df630e5

File tree

3 files changed

+18
-9
lines changed

3 files changed

+18
-9
lines changed

src/pycbsdk/cbhw/device/nsp.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -346,9 +346,13 @@ def _register_basic_callbacks(self):
346346

347347
self.register_config_callback(CBPacketType.CHANREPAINP, self._handle_chaninfo)
348348
self.register_config_callback(CBPacketType.CHANREPSPKTHR, self._handle_chaninfo)
349-
self.register_config_callback(CBPacketType.CHANREPNTRODEGROUP, self._handle_chaninfo)
349+
self.register_config_callback(
350+
CBPacketType.CHANREPNTRODEGROUP, self._handle_chaninfo
351+
)
350352
self.register_config_callback(CBPacketType.CHANREPDISP, self._handle_chaninfo)
351-
self.register_config_callback(CBPacketType.CHANREPUNITOVERRIDES, self._handle_chaninfo)
353+
self.register_config_callback(
354+
CBPacketType.CHANREPUNITOVERRIDES, self._handle_chaninfo
355+
)
352356

353357
self.register_config_callback(CBPacketType.GROUPREP, self._handle_groupinfo)
354358
self.register_config_callback(CBPacketType.PROCREP, self._handle_procinfo)
@@ -400,7 +404,9 @@ def _handle_sysrep(self, pkt):
400404
def _handle_chaninfo(self, pkt):
401405
# If this config packet is limited in scope then it might have some garbage data in its out-of-scope payload.
402406
# We should update our config, but only the parts that this REP packet is scoped to.
403-
if (pkt.header.instrument != self._config["instrument"]) or (pkt.chan > self._config["proc_chans"]):
407+
if (pkt.header.instrument != self._config["instrument"]) or (
408+
pkt.chan > self._config["proc_chans"]
409+
):
404410
# Drop channels that do not belong to this instrument
405411
pass
406412
elif pkt.header.type in [CBPacketType.CHANREP]:
@@ -446,7 +452,10 @@ def _handle_chaninfo(self, pkt):
446452
elif pkt.header.type == CBPacketType.CHANREPLABEL:
447453
self._config["channel_infos"][pkt.chan].label = pkt.label
448454
self._config["channel_infos"][pkt.chan].userflags = pkt.userflags
449-
elif pkt.header.type in [CBPacketType.CHANSETSPKTHR, CBPacketType.CHANREPSPKTHR]:
455+
elif pkt.header.type in [
456+
CBPacketType.CHANSETSPKTHR,
457+
CBPacketType.CHANREPSPKTHR,
458+
]:
450459
self._config["channel_infos"][pkt.chan].spkthrlevel = pkt.spkthrlevel
451460
elif pkt.header.type == CBPacketType.CHANREPNTRODEGROUP:
452461
# TODO: from use pkt.spkgroup
@@ -502,7 +511,7 @@ def _handle_procmon(self, pkt):
502511
f";\tcounter - {pkt.counter if has_counter else 'N/A'}"
503512
f";\tdelta - {pkt_delta}"
504513
f";\tsent - {pkt.sentpkts}"
505-
f";\trate (pkt/samp) - {pkt_delta/update_interval}"
514+
f";\trate (pkt/samp) - {pkt_delta / update_interval}"
506515
)
507516
self._monitor_state = {
508517
"counter": pkt.counter if has_counter else -1,

src/pycbsdk/cbhw/packet/packets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -530,7 +530,7 @@ def sizes(self) -> list[int]:
530530
# Convert coords from uint16 to half as many uint32
531531
# TODO: Use numpy from buffer
532532
return struct.unpack(
533-
f"<{len(self.coords)//2}L",
533+
f"<{len(self.coords) // 2}L",
534534
struct.pack(f"<{len(self.coords)}H", self._array),
535535
)
536536

@@ -540,7 +540,7 @@ def sizes(self, insizes: list[int]):
540540
assert n_elems <= (self.max_elements // 2)
541541
# TODO: Use numpy buffer
542542
self.coords = struct.unpack(
543-
f"<{len(insizes)*2}H", struct.pack(f"<{len(insizes)}L", insizes)
543+
f"<{len(insizes) * 2}H", struct.pack(f"<{len(insizes)}L", insizes)
544544
)
545545

546546
@property

src/pycbsdk/examples/group_sample_intervals.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(self, nchans: int, duration=21.0, t_step=1 / 30_000):
2323

2424
def handle_frame(self, pkt):
2525
if self._write_index < self._buffer.shape[0]:
26-
self._buffer[self._write_index, :] = memoryview(pkt.data[:self._nchans])
26+
self._buffer[self._write_index, :] = memoryview(pkt.data[: self._nchans])
2727
self._ts[self._write_index] = pkt.header.time
2828
self._write_index += 1
2929

@@ -35,7 +35,7 @@ def finish(self):
3535
s_elapsed = ts_elapsed * self._t_step
3636
n_samps = np.sum(b_ts)
3737
print(
38-
f"Collected {n_samps} samples in {s_elapsed} s\t({n_samps/s_elapsed:.2f} Hz)."
38+
f"Collected {n_samps} samples in {s_elapsed} s\t({n_samps / s_elapsed:.2f} Hz)."
3939
)
4040

4141

0 commit comments

Comments
 (0)