xrtm-data 0.2.1__tar.gz → 0.2.3__tar.gz

{xrtm_data-0.2.1 → xrtm_data-0.2.3}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: xrtm-data
-Version: 0.2.1
+Version: 0.2.3
 Summary: The Snapshot Vault for XRTM.
 Author-email: XRTM Team <moy@xrtm.org>
-License: Apache-2.0
+License-Expression: Apache-2.0
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE

{xrtm_data-0.2.1 → xrtm_data-0.2.3}/pyproject.toml

@@ -4,11 +4,11 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "xrtm-data"
-version = "0.2.1"
+version = "0.2.3"
 description = "The Snapshot Vault for XRTM."
 readme = "README.md"
 requires-python = ">=3.11"
-license = {text = "Apache-2.0"}
+license = "Apache-2.0"
 authors = [
     {name = "XRTM Team", email = "moy@xrtm.org"}
 ]
@@ -38,7 +38,6 @@ package-dir = {"" = "src"}
 packages = {find = {where = ["src"], include = ["xrtm*"], namespaces = true}}
 
 [tool.pytest.ini_options]
-pythonpath = ["src"]
 testpaths = ["tests"]
 asyncio_mode = "strict"
 

{xrtm_data-0.2.1 → xrtm_data-0.2.3}/src/xrtm/data/cli/__init__.py

@@ -75,6 +75,9 @@ def collect(
         xrtm-data collect -m 0x1234... -d 30 -o data/trades.parquet
     """
     output_path = Path(output)
+    output_suffix = output_path.suffix.lower()
+    if output_suffix not in {".parquet", ".json"}:
+        raise click.ClickException("Output path must end with .parquet or .json")
 
     # Check cache
     if output_path.exists() and not force:
@@ -124,10 +127,12 @@ def collect(
     # Save output
     output_path.parent.mkdir(parents=True, exist_ok=True)
 
-    if output_path.suffix == ".parquet":
+    if output_suffix == ".parquet":
         _save_parquet(window, output_path)
-    else:
+    elif output_suffix == ".json":
         _save_json(window, output_path)
+    else:  # pragma: no cover - guarded before collection
+        raise click.ClickException("Output path must end with .parquet or .json")
 
     console.print(f"[green]✓ Saved {len(window.trades)} trades to:[/green] {output_path}")
 
@@ -214,8 +219,9 @@ def info(file_path: str):
     Displays summary statistics for trade files or prior files.
     """
     path = Path(file_path)
+    suffix = path.suffix.lower()
 
-    if path.suffix == ".json":
+    if suffix == ".json":
         with open(path) as f:
             data = json.load(f)
 
@@ -230,7 +236,7 @@ def info(file_path: str):
             ))
         else:
             console.print(f"JSON file with {len(data)} keys")
-    elif path.suffix == ".parquet":
+    elif suffix == ".parquet":
         import pyarrow.parquet as pq
 
         table = pq.read_table(path)
@@ -240,6 +246,8 @@ def info(file_path: str):
             f"Columns: {table.column_names}",
             title=path.name,
         ))
+    else:
+        raise click.ClickException("Unsupported file type. Expected .json or .parquet")
 
 
 def _save_parquet(window, path: Path) -> None:
@@ -289,7 +297,6 @@ def _load_trades(path: Path) -> list:
         import pyarrow.parquet as pq
 
         table = pq.read_table(path)
-        df = table.to_pandas()
         return [
             TradeEvent(
                 price=row["price"],
@@ -298,7 +305,7 @@ def _load_trades(path: Path) -> list:
                 maker=row["maker"],
                 taker=row["taker"],
             )
-            for _, row in df.iterrows()
+            for row in table.to_pylist()
         ]
     else:
         with open(path) as f:

{xrtm_data-0.2.1 → xrtm_data-0.2.3}/src/xrtm/data/core/schemas/forecast.py

@@ -28,7 +28,7 @@ Example:
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Optional
 
-from pydantic import AliasChoices, BaseModel, ConfigDict, Field
+from pydantic import AliasChoices, BaseModel, ConfigDict, Field, model_validator
 
 
 class MetadataBase(BaseModel):
@@ -202,6 +202,41 @@ class ForecastOutput(BaseModel):
     calibration_metrics: Dict[str, Any] = Field(default_factory=dict, description="Performance metrics")
     metadata: MetadataBase = Field(default_factory=MetadataBase)  # type: ignore[arg-type]
 
+    @model_validator(mode="before")
+    @classmethod
+    def _apply_reasoning_trace_alias(cls, data: Any) -> Any:
+        r"""Accept governance ``reasoning_trace`` as an alias for runtime trace fields."""
+        if not isinstance(data, dict) or "reasoning_trace" not in data:
+            return data
+
+        trace = data["reasoning_trace"]
+        updated = dict(data)
+        if isinstance(trace, dict):
+            if "reasoning" not in updated and isinstance(trace.get("narrative"), str):
+                updated["reasoning"] = trace["narrative"]
+
+            causal_graph = trace.get("causal_graph")
+            if isinstance(causal_graph, dict):
+                if "logical_trace" not in updated and "nodes" in causal_graph:
+                    updated["logical_trace"] = causal_graph["nodes"]
+                if "logical_edges" not in updated and "edges" in causal_graph:
+                    updated["logical_edges"] = causal_graph["edges"]
+        elif isinstance(trace, list) and "logical_trace" not in updated:
+            updated["logical_trace"] = trace
+
+        return updated
+
+    @property
+    def reasoning_trace(self) -> Dict[str, Any]:
+        r"""Governance-compatible alias for the narrative and causal graph trace."""
+        return {
+            "narrative": self.reasoning,
+            "causal_graph": {
+                "nodes": [node.model_dump(exclude_none=True) for node in self.logical_trace],
+                "edges": [edge.model_dump(exclude_none=True) for edge in self.logical_edges],
+            },
+        }
+
     @property
     def confidence(self) -> float:
         r"""Backward compatibility alias for probability."""
@@ -227,15 +262,17 @@ class ForecastOutput(BaseModel):
         except ImportError:
             raise ImportError("networkx is required for to_networkx(). Install it with 'uv add networkx'.")
         dg = nx.DiGraph()
-        for node in self.logical_trace:
-            dg.add_node(
+        dg.add_nodes_from(
+            (
                 node.node_id,
-                event=node.event,
-                probability=node.probability,
-                description=node.description,
+                {"event": node.event, "probability": node.probability, "description": node.description},
             )
-        for edge in self.logical_edges:
-            dg.add_edge(edge.source, edge.target, weight=edge.weight, description=edge.description)
+            for node in self.logical_trace
+        )
+        dg.add_edges_from(
+            (edge.source, edge.target, {"weight": edge.weight, "description": edge.description})
+            for edge in self.logical_edges
+        )
         return dg
 
 

xrtm_data-0.2.3/src/xrtm/data/corpora/__init__.py

@@ -0,0 +1,36 @@
+# coding=utf-8
+# Copyright 2026 XRTM Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+r"""Deterministic fixture corpora for offline validation."""
+
+from xrtm.data.corpora.real_binary import (
+    REAL_BINARY_CORPUS_ID,
+    RealBinaryCorpusSource,
+    RealBinaryQuestionRecord,
+    load_real_binary_corpus,
+    load_real_binary_questions,
+    load_real_binary_resolved_outcomes,
+    validate_real_binary_corpus,
+)
+
+__all__ = [
+    "REAL_BINARY_CORPUS_ID",
+    "RealBinaryQuestionRecord",
+    "RealBinaryCorpusSource",
+    "load_real_binary_corpus",
+    "load_real_binary_questions",
+    "load_real_binary_resolved_outcomes",
+    "validate_real_binary_corpus",
+]