trainfabric 0.1.15 → 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.15";
+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");

package/package.json

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