{"record":{"id":"06a461456d111b71","repo":"keras-team/keras","slug":"argument-num-segments-must-be-set-when-using-the-06a461","errorCode":null,"errorMessage":"Argument `num_segments` must be set when using the JAX backend. Received: num_segments=None","messagePattern":"Argument `num_segments` must be set when using the JAX backend\\. Received: num_segments=None","errorType":"exception","errorClass":"ValueError","httpStatus":null,"severity":"error","filePath":"keras/src/backend/jax/ops/math.py","lineNumber":13,"sourceCode":"import math\n\nimport jax\nimport jax.numpy as jnp\n\nfrom keras.src.backend.common.variables import standardize_dtype\nfrom keras.src.backend.jax.ops.core import convert_to_tensor\nfrom keras.src.utils.module_utils import scipy\n\n\ndef segment_sum(data, segment_ids, num_segments=None, sorted=False):\n    if num_segments is None:\n        raise ValueError(\n            \"Argument `num_segments` must be set when using the JAX backend. \"\n            \"Received: num_segments=None\"\n        )\n    return jax.ops.segment_sum(\n        data, segment_ids, num_segments, indices_are_sorted=sorted\n    )\n\n\ndef segment_max(data, segment_ids, num_segments=None, sorted=False):\n    if num_segments is None:\n        raise ValueError(\n            \"Argument `num_segments` must be set when using the JAX backend. \"\n            \"Received: num_segments=None\"\n        )\n    return jax.ops.segment_max(\n        data, segment_ids, num_segments, indices_are_sorted=sorted\n    )\n","sourceCodeStart":1,"sourceCodeEnd":31,"githubUrl":"https://github.com/keras-team/keras/blob/7a34a03db60bf60042242d6a556fc3be119046a5/keras/src/backend/jax/ops/math.py#L1-L31","documentation":"Error \"Argument `num_segments` must be set when using the JAX backend. Received: num_segments=None\" thrown in keras-team/keras.","triggerScenarios":"Thrown at keras/src/backend/jax/ops/math.py:13 when the library encounters an invalid state.","commonSituations":"See trigger scenarios.","solutions":[],"exampleFix":null,"handlingStrategy":null,"validationCode":null,"typeGuard":null,"tryCatchPattern":null,"preventionTips":[],"tags":[],"backgroundTag":null,"analyzedSha":"7a34a03db60bf60042242d6a556fc3be119046a5","analyzedAt":"2026-08-25T21:25:25.994Z","schemaVersion":2},"datasetVersion":"2026-08-26T02:17:13.382Z"}