def remove(self, elem):
        """Remove a requirement.

        :param elem: a string or :class:`Requirement` instance.
        """
        if isinstance(elem, text_type):
            for requirement in self:
                if requirement.package == elem:
                    return super(Requirements, self).remove(requirement)
        return super(Requirements, self).remove(elem)
def test_telemetry_imports_jax(runner, live_mock_server, parse_ctx):
    with runner.isolated_filesystem():
        import jax

        wandb.init()
        wandb.finish()

        ctx_util = parse_ctx(live_mock_server.get_ctx())
        telemetry = ctx_util.telemetry

        # jax in finish modules but not in init modules
        assert telemetry and 12 in telemetry.get("1", [])
        assert telemetry and 12 in telemetry.get("2", [])