@neuracore/types 6.13.0 → 7.1.0

package/dist/neuracore_types.d.ts

@@ -81,6 +81,30 @@ export interface BaseRecodingUpdatePayload {
     robot_id: string;
     instance: number;
 }
+/**
+ * Request body for updating the status/progress of multiple data traces.
+ *
+ * Attributes:
+ *     updates: A dictionary of trace_id -> TraceStatusUpdates
+ */
+export interface BatchedUpdateTraceRequest {
+    updates: {
+        [k: string]: TraceStatusUpdates;
+    };
+}
+/**
+ * Represents an update status/progress of a data trace.
+ *
+ * Attributes:
+ *     status: The status of the trace upload.
+ *     uploaded_bytes: The number of bytes uploaded.
+ *     total_bytes: The total number of bytes expected.
+ */
+export interface TraceStatusUpdates {
+    status?: RecordingDataTraceStatus | null;
+    uploaded_bytes?: number | null;
+    total_bytes?: number | null;
+}
 /**
  * Camera sensor data including images and calibration information.
  *
@@ -780,19 +804,25 @@ export interface DeploymentConfig {
  *     training_id: Identifier of the trained model to deploy.
  *     name: Optional name for the endpoint.
  *     ttl: Optional time-to-live in seconds for the endpoint.
- *     model_input_order: Specification of the model input data order.
- *     model_output_order: Specification of the model output data order.
+ *     input_embodiment_description: Specification of the model input data
+ *         order.
+ *     output_embodiment_description: Specification of the model output data
+ *         order.
  *     config: Deployment configuration parameters.
  */
 export interface DeploymentRequest {
     training_id: string;
     name?: string | null;
     ttl?: number | null;
-    model_input_order: {
-        [k: string]: string[];
+    input_embodiment_description: {
+        [k: string]: {
+            [k: string]: string;
+        };
     };
-    model_output_order: {
-        [k: string]: string[];
+    output_embodiment_description: {
+        [k: string]: {
+            [k: string]: string;
+        };
     };
     config?: DeploymentConfig;
 }
@@ -912,8 +942,10 @@ export interface HandshakeMessage {
  *     algorithm_config: Configuration parameters for the algorithm.
  *     gpu_type: The type of GPU to use for the job.
  *     num_gpus: The number of GPUs to allocate for the job.
- *     input_robot_data_spec: Mapping of robot to input data types to names
- *     output_robot_data_spec: Mapping of robot to output data types to names
+ *     input_cross_embodiment_description: Mapping of robot to input data
+ *         types to names
+ *     output_cross_embodiment_description: Mapping of robot to output data
+ *         types to names
  */
 export interface InternalStartTrainingJobRequest {
     org_id: string;
@@ -928,14 +960,18 @@ export interface InternalStartTrainingJobRequest {
     gpu_type: GPUType;
     num_gpus: number;
     synchronization_details: SynchronizationDetails;
-    input_robot_data_spec: {
+    input_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
-    output_robot_data_spec: {
+    output_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
 }
@@ -944,7 +980,8 @@ export interface InternalStartTrainingJobRequest {
  *
  * Attributes:
  *     frequency: Synchronization frequency in Hz.
- *     robot_data_spec: Specification of robot data to include in the synchronization.
+ *     cross_embodiment_description: Specification of robot data to include
+ *         in the synchronization.
  *     max_delay_s: Maximum allowable delay (in seconds) for synchronization.
  *     allow_duplicates: Whether to allow duplicate data points in the synchronization.
  *     trim_start_end: Whether to trim the start and end of the episode
@@ -952,7 +989,7 @@ export interface InternalStartTrainingJobRequest {
  */
 export interface SynchronizationDetails {
     frequency: number;
-    robot_data_spec: {
+    cross_embodiment_union: {
         [k: string]: {
             [k: string]: string[];
         };
@@ -1032,9 +1069,11 @@ export interface MetricsData {
  *     ModelInitDescription(
  *         input_data_types=[DataType.RGB_IMAGES, DataType.JOINT_POSITIONS],
  *         output_data_types=[DataType.JOINT_TARGET_POSITIONS],
- *         dataset_statistics={
+ *         input_dataset_statistics={
  *             DataType.RGB_IMAGES: DataItemStats(...),
  *             DataType.JOINT_POSITIONS: DataItemStats(...),
+ *         },
+ *         output_dataset_statistics={
  *             DataType.JOINT_TARGET_POSITIONS: DataItemStats(...),
  *         },
  *     )
@@ -1042,7 +1081,10 @@ export interface MetricsData {
 export interface ModelInitDescription {
     input_data_types: DataType[];
     output_data_types: DataType[];
-    dataset_statistics: {
+    input_dataset_statistics: {
+        [k: string]: (JointDataStats | CameraDataStats | PoseDataStats | EndEffectorPoseDataStats | ParallelGripperOpenAmountDataStats | PointCloudDataStats | LanguageDataStats | Custom1DDataStats)[];
+    };
+    output_dataset_statistics: {
         [k: string]: (JointDataStats | CameraDataStats | PoseDataStats | EndEffectorPoseDataStats | ParallelGripperOpenAmountDataStats | PointCloudDataStats | LanguageDataStats | Custom1DDataStats)[];
     };
     output_prediction_horizon: number;
@@ -1287,7 +1329,7 @@ export interface RecordingDataTrace {
     id: string;
     recording_id: string;
     data_type: DataType;
-    status: RecordingDataTraceStatus;
+    status: RecordingDataTraceStatus1;
     created_at: number;
     uploaded_at: number | null;
     uploaded_bytes: number;
@@ -1421,18 +1463,29 @@ export interface SynchronizedDataset {
  *
  * Attributes:
  *     synchronized_dataset_id: Unique identifier for the synced dataset.
- *     robot_data_spec: Mapping of robot IDs to data type names.
+ *     cross_embodiment_description: Mapping of robot IDs to data type names.
  *     dataset_statistics: Statistics for each robot and data type.
  */
 export interface SynchronizedDatasetStatistics {
     synchronized_dataset_id: string;
-    robot_data_spec: {
+    input_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
+        };
+    };
+    output_cross_embodiment_description: {
+        [k: string]: {
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
     dataset_statistics: {
-        [k: string]: (JointDataStats | CameraDataStats | PoseDataStats | EndEffectorPoseDataStats | ParallelGripperOpenAmountDataStats | PointCloudDataStats | LanguageDataStats | Custom1DDataStats)[];
+        [k: string]: {
+            [k: string]: (JointDataStats | CameraDataStats | PoseDataStats | EndEffectorPoseDataStats | ParallelGripperOpenAmountDataStats | PointCloudDataStats | LanguageDataStats | Custom1DDataStats)[];
+        };
     };
 }
 /**
@@ -1496,8 +1549,8 @@ export interface TracesMetadataRequest {
  *     previous_training_time: The time spent on the previous training, if applicable.
  *     error: Any error message associated with the job, if applicable.
  *     resume_points: List of timestamps where the job can be resumed.
- *     input_robot_data_spec: List of data types for the input data.
- *     output_robot_data_spec: List of data types for the output data.
+ *     input_cross_embodiment_description: List of data types for the input data.
+ *     output_cross_embodiment_description: List of data types for the output data.
  */
 export interface TrainingJob {
     id: string;
@@ -1523,14 +1576,18 @@ export interface TrainingJob {
     previous_training_time?: number | null;
     error?: string | null;
     resume_points: number[];
-    input_robot_data_spec: {
+    input_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
-    output_robot_data_spec: {
+    output_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
     synchronization_details: SynchronizationDetails;
@@ -1546,9 +1603,9 @@ export interface TrainingJob {
  *     frequency: Synchronization frequency in Hz.
  *     gpu_type: Type of GPU to use for training.
  *     num_gpus: Number of GPUs to use for training.
- *     input_robot_data_spec: The robust mapping of robot to input
+ *     input_cross_embodiment_description: The robust mapping of robot to input
  *         data types to names
- *     output_robot_data_spec: The robust mapping of robot to output
+ *     output_cross_embodiment_description: The robust mapping of robot to output
  *         data types to names
  */
 export interface TrainingJobRequest {
@@ -1561,14 +1618,18 @@ export interface TrainingJobRequest {
     gpu_type: GPUType;
     num_gpus: number;
     synchronization_details: SynchronizationDetails;
-    input_robot_data_spec: {
+    input_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
-    output_robot_data_spec: {
+    output_cross_embodiment_description: {
         [k: string]: {
-            [k: string]: string[];
+            [k: string]: {
+                [k: string]: string;
+            };
         };
     };
 }
@@ -1594,6 +1655,14 @@ export declare enum DataType {
     LANGUAGE = "LANGUAGE",
     CUSTOM_1D = "CUSTOM_1D"
 }
+/**
+ * Status for a recording data trace upload lifecycle.
+ */
+export declare enum RecordingDataTraceStatus {
+    QUEUED = "QUEUED",
+    UPLOAD_STARTED = "UPLOAD_STARTED",
+    UPLOAD_COMPLETE = "UPLOAD_COMPLETE"
+}
 export declare enum Type {
     CameraDataStats = "CameraDataStats"
 }
@@ -1879,7 +1948,7 @@ export declare enum Type16 {
 /**
  * Status for a recording data trace upload lifecycle.
  */
-export declare enum RecordingDataTraceStatus {
+export declare enum RecordingDataTraceStatus1 {
     QUEUED = "QUEUED",
     UPLOAD_STARTED = "UPLOAD_STARTED",
     UPLOAD_COMPLETE = "UPLOAD_COMPLETE"

package/dist/neuracore_types.js

@@ -6,7 +6,7 @@
 /* Do not modify it by hand - just update the pydantic models and then re-run the script
 */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GPUType2 = exports.TrainingJobStatus = exports.RecordingNotificationType = exports.RecordingDataTraceStatus = exports.Type16 = exports.Type15 = exports.Type14 = exports.PendingRecordingStatus = exports.RecordingStatus = exports.Type13 = exports.VideoFormat = exports.Type12 = exports.Type11 = exports.Type10 = exports.Type9 = exports.Type8 = exports.Type7 = exports.Type6 = exports.MessageType = exports.GPUType1 = exports.EndpointStatus = exports.Type5 = exports.Type4 = exports.Type3 = exports.GPUType = exports.DatasetTypeConfig = exports.Type2 = exports.IntrinsicsConfig = exports.PoseConfig1 = exports.ActionSpaceConfig = exports.ActionTypeConfig = exports.JointPositionInputTypeConfig = exports.VisualJointInputTypeConfig = exports.LanguageConfig = exports.EndEffectorPoseInputTypeConfig = exports.AngleConfig1 = exports.EulerOrderConfig = exports.QuaternionOrderConfig = exports.RotationConfig = exports.PoseConfig = exports.DistanceUnitsConfig = exports.TorqueUnitsConfig = exports.AngleConfig = exports.ImageChannelOrderConfig = exports.ImageConventionConfig = exports.Type1 = exports.Type = exports.DataType = void 0;
+exports.GPUType2 = exports.TrainingJobStatus = exports.RecordingNotificationType = exports.RecordingDataTraceStatus1 = exports.Type16 = exports.Type15 = exports.Type14 = exports.PendingRecordingStatus = exports.RecordingStatus = exports.Type13 = exports.VideoFormat = exports.Type12 = exports.Type11 = exports.Type10 = exports.Type9 = exports.Type8 = exports.Type7 = exports.Type6 = exports.MessageType = exports.GPUType1 = exports.EndpointStatus = exports.Type5 = exports.Type4 = exports.Type3 = exports.GPUType = exports.DatasetTypeConfig = exports.Type2 = exports.IntrinsicsConfig = exports.PoseConfig1 = exports.ActionSpaceConfig = exports.ActionTypeConfig = exports.JointPositionInputTypeConfig = exports.VisualJointInputTypeConfig = exports.LanguageConfig = exports.EndEffectorPoseInputTypeConfig = exports.AngleConfig1 = exports.EulerOrderConfig = exports.QuaternionOrderConfig = exports.RotationConfig = exports.PoseConfig = exports.DistanceUnitsConfig = exports.TorqueUnitsConfig = exports.AngleConfig = exports.ImageChannelOrderConfig = exports.ImageConventionConfig = exports.Type1 = exports.Type = exports.RecordingDataTraceStatus = exports.DataType = void 0;
 /**
  * Enumeration of supported data types in the Neuracore system.
  *
@@ -30,6 +30,15 @@ var DataType;
     DataType["LANGUAGE"] = "LANGUAGE";
     DataType["CUSTOM_1D"] = "CUSTOM_1D";
 })(DataType || (exports.DataType = DataType = {}));
+/**
+ * Status for a recording data trace upload lifecycle.
+ */
+var RecordingDataTraceStatus;
+(function (RecordingDataTraceStatus) {
+    RecordingDataTraceStatus["QUEUED"] = "QUEUED";
+    RecordingDataTraceStatus["UPLOAD_STARTED"] = "UPLOAD_STARTED";
+    RecordingDataTraceStatus["UPLOAD_COMPLETE"] = "UPLOAD_COMPLETE";
+})(RecordingDataTraceStatus || (exports.RecordingDataTraceStatus = RecordingDataTraceStatus = {}));
 var Type;
 (function (Type) {
     Type["CameraDataStats"] = "CameraDataStats";
@@ -358,12 +367,12 @@ var Type16;
 /**
  * Status for a recording data trace upload lifecycle.
  */
-var RecordingDataTraceStatus;
-(function (RecordingDataTraceStatus) {
-    RecordingDataTraceStatus["QUEUED"] = "QUEUED";
-    RecordingDataTraceStatus["UPLOAD_STARTED"] = "UPLOAD_STARTED";
-    RecordingDataTraceStatus["UPLOAD_COMPLETE"] = "UPLOAD_COMPLETE";
-})(RecordingDataTraceStatus || (exports.RecordingDataTraceStatus = RecordingDataTraceStatus = {}));
+var RecordingDataTraceStatus1;
+(function (RecordingDataTraceStatus1) {
+    RecordingDataTraceStatus1["QUEUED"] = "QUEUED";
+    RecordingDataTraceStatus1["UPLOAD_STARTED"] = "UPLOAD_STARTED";
+    RecordingDataTraceStatus1["UPLOAD_COMPLETE"] = "UPLOAD_COMPLETE";
+})(RecordingDataTraceStatus1 || (exports.RecordingDataTraceStatus1 = RecordingDataTraceStatus1 = {}));
 /**
  * Types of recording lifecycle notifications.
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neuracore/types",
-  "version": "6.13.0",
+  "version": "7.1.0",
   "description": "Shared TypeScript type definitions for Neuracore robotics platform",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",