Cant use NdxEventsNWBFile with normal NWBFile functions

[Lathomas42]

nwbfile.units.get_ancestor(data_type='NWBFile') returns None, so functions like "nwbfile.add_unit" fails when using a NdxEventsNWBFile

to reproduce, just try to add a unit after running the write_ttls_events.py script:
nwbfile.add_unit(spike_times=[0,1])

Let me know if I can provide any other information. Id love to use the events table, but cannot if it means I cannot use the rest of the nwb functions. Let me know if there is a workaround or somethign I am doing wrong.

[Lathomas42]

Sorry, found the specifics of what causes this issue.

Trying to add a unit with an electrode associated with it. Here is a minimum code to recreate the issue:

from datetime import datetime
from pynwb import NWBHDF5IO

from ndx_events import (
    EventsTable,
    CategoricalVectorData,
    MeaningsTable,
    NdxEventsNWBFile,
)

nwbfile = NdxEventsNWBFile(
    session_description="session description",
    identifier="cool_experiment_001",
    session_start_time=datetime.now().astimezone(),
)
electrode_group = nwbfile.create_electrode_group(
                name='test_group',
                description="test",
                location='unknown',
                device=nwbfile.create_device(name='test_device'))


nwbfile.add_electrode(
        id = 0,
        group=electrode_group,
        location="brain area"
    )

nwbfile.add_unit(spike_times=[0,1],electrodes=[0])

The error is then:

AttributeError: 'NoneType' object has no attribute 'electrodes'

and it results from the fact that

nwbfile.units.get_ancestor(data_type='NWBFile') is None