trainfabric 0.1.14 → 0.1.16

package/dist/index.cjs

@@ -8799,11 +8799,16 @@ var import_node_fs = __toESM(require("node:fs"), 1);
 var import_node_os = __toESM(require("node:os"), 1);
 var import_node_path2 = __toESM(require("node:path"), 1);
 function collectRuntimeFiles(repoPath) {
-  const candidates = ["training.yaml", "train.runtime.yaml", "pyproject.toml", "requirements.txt", "environment.yml", "Dockerfile", "train.py"];
+  const manifestFiles = /* @__PURE__ */ new Set(["training.yaml", "train.runtime.yaml", "pyproject.toml", "requirements.txt", "environment.yml", "Dockerfile"]);
   const ignoredDirectories = /* @__PURE__ */ new Set([".git", ".hg", ".svn", "node_modules", ".venv", "venv", "__pycache__", ".mypy_cache", ".pytest_cache", "dist", "build"]);
   const found = [];
+  const maxFiles = 256;
+  const maxFileBytes = 256 * 1024;
+  function shouldIncludeFile(fileName) {
+    return manifestFiles.has(fileName) || fileName.endsWith(".py");
+  }
   function walk(currentPath) {
-    if (found.length >= 64) {
+    if (found.length >= maxFiles) {
       return;
     }
     for (const entry of import_node_fs.default.readdirSync(currentPath, { withFileTypes: true })) {
@@ -8815,11 +8820,11 @@ function collectRuntimeFiles(repoPath) {
         }
         continue;
       }
-      if (!entry.isFile() || !candidates.includes(entry.name)) {
+      if (!entry.isFile() || !shouldIncludeFile(entry.name)) {
         continue;
       }
       const stat = import_node_fs.default.statSync(absolute);
-      if (stat.size > 256 * 1024) {
+      if (stat.size > maxFileBytes) {
         continue;
       }
       found.push({
@@ -8922,7 +8927,7 @@ function buildComputeSpec(options) {
 
 // src/index.ts
 var DEFAULT_TRAINFABRIC_API_URL2 = "https://api.trainfabric.com";
-var CLI_VERSION = "0.1.14";
+var CLI_VERSION = "0.1.16";
 var CONFIG_DIR = import_node_path3.default.join(import_node_os2.default.homedir(), ".trainfabric");
 var CONFIG_PATH = import_node_path3.default.join(CONFIG_DIR, "config.json");
 var FALLBACK_SECRET_PATH = import_node_path3.default.join(CONFIG_DIR, "session.enc");
@@ -9292,6 +9297,9 @@ function assertQuoteOptions(bundle, options) {
     return;
   }
   const constraints = [];
+  if (options.mode) {
+    constraints.push(`mode=${options.mode}`);
+  }
   if (options.accelerator) {
     constraints.push(`accelerator=${options.accelerator}`);
   }
@@ -9304,7 +9312,7 @@ function assertQuoteOptions(bundle, options) {
   throw new Error(
     `No quote options were returned for the requested run under the current hardware constraints (${constraints.join(
       ", "
-    )}). Omit --accelerator or increase --min-memory.`
+    )}). Try a different --mode, omit --accelerator, adjust --min-memory if it excludes available GPUs, or retry when capacity changes.`
   );
 }
 function filterQuoteBundleForRequestedMode(bundle, options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "trainfabric",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "description": "Trainfabric CLI for launching GPU training jobs on the hosted Trainfabric backend.",
   "type": "module",
   "main": "dist/index.cjs",