fix weights file name check

CIROH-UA · Jan 30, 2024 · b708346 · b708346
1 parent e2557fa
commit b708346
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/stream.sh b/scripts/stream.sh
@@ -216,7 +216,7 @@ build_docker_container "$DOCKER_TAG" "$FP_DOCKER"
 WEIGHTS_FILENAME=$(find "$DATASTREAM_RESOURCES" -type f -name "*weights*")
 if [ -e "$WEIGHTS_FILENAME" ]; then
     echo "Using weights found in resources directory $WEIGHTS_FILENAME"
-    if [[ "$WEIGHTS_FILENAME" != "*/weights.json" ]]; then
+    if [[ $(basename $WEIGHTS_FILENAME) != "weights.json" ]]; then
         mv "$WEIGHTS_FILENAME" ""$DATASTREAM_RESOURCES"/weights.json"
     fi
 else