Skip to content

Commit 74b3032

Browse files
Update requirements and minor edits to run nsvs
1 parent 3fc66d8 commit 74b3032

File tree

3 files changed

+30
-27
lines changed

3 files changed

+30
-27
lines changed

ns_vfs/api/run_with_nsvqa.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def run_nsvs_nsvqa(
102102

103103

104104
if __name__ == "__main__":
105-
input_data_path = "/nas/mars/experiment_result/nsvqa/1_puls/longvideobench/longvideobench-outputs-updated.json"
105+
input_data_path = "/nas/mars/experiment_result/nsvqa/1_puls/longvideobench/longvideobench-outputs-fixed-specs-v2.json"
106106
with open(input_data_path, 'r', encoding='utf-8') as f:
107107
data = json.load(f)
108108

ns_vfs/dataloader/_base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
from typing import List, Dict, Union
22
import abc
3+
import numpy as np
34

45
class DatasetLoader(abc.ABC):
56
"""Base class for loading dataset."""

pyproject.toml

Lines changed: 28 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -3,44 +3,47 @@ requires = ["setuptools", "wheel"]
33
build-backend = "setuptools.build_meta"
44

55
[tool.setuptools.packages.find]
6-
where = ["."] # list of folders that contain the packages (["."] by default)
7-
include = ["ns_vfs"] # package names should match these glob patterns (["*"] by default)
8-
exclude = ["store", "artifacts, docker, experiments"] # exclude packages matching these glob patterns (empty by default)
9-
namespaces = false # to disable scanning PEP 420 namespaces (true by default)
6+
where = ["."] # list of folders that contain the packages (["."] by default)
7+
include = [
8+
"ns_vfs",
9+
] # package names should match these glob patterns (["*"] by default)
10+
exclude = [
11+
"store",
12+
"artifacts, docker, experiments",
13+
] # exclude packages matching these glob patterns (empty by default)
14+
namespaces = false # to disable scanning PEP 420 namespaces (true by default)
1015

1116
[tool.my-tool]
1217
submodule_path = "ns_vfs/model/diffusion/stable_diffusion"
1318

1419
[project]
1520
name = "ns-vfs"
1621
version = "0.0.1"
17-
authors = [
18-
{ name="Minkyu Choi", email="[email protected]" },
19-
]
22+
authors = [{ name = "Minkyu Choi", email = "[email protected]" }]
2023
description = "The pipeline to convert video to automaton"
2124
readme = "README.md"
2225
requires-python = ">=3.10"
2326
classifiers = [
24-
"Programming Language :: Python :: 3",
25-
"License :: OSI Approved :: MIT License",
26-
"Operating System :: OS Independent",
27+
"Programming Language :: Python :: 3",
28+
"License :: OSI Approved :: MIT License",
29+
"Operating System :: OS Independent",
2730
]
2831
dependencies = [
29-
"omegaconf",
30-
"hydra-core>=1.2.0",
31-
"numpy==1.21.5",
32-
"pandas==2.0.3",
33-
"sympy",
34-
"regex",
35-
"supervision==0.6.0",
36-
"roboflow",
37-
"einops",
38-
"pytorch-lightning==1.4.2",
39-
"torchmetrics==0.5",
40-
"ultralytics",
41-
"clip@git+https://github.com/openai/CLIP.git",
42-
"cvias@git+https://github.com/Coargus/Computer-Vision-Inference-and-Server.git@nsvqa-dev",
43-
"vflow@git+https://github.com/Coargus/[email protected]",
32+
"omegaconf==2.3.0",
33+
"hydra-core>=1.3.2",
34+
"numpy==1.26.4",
35+
"pandas==2.0.3",
36+
"sympy==1.13.1",
37+
"regex==2024.11.6",
38+
"supervision==0.6.0",
39+
"roboflow==1.1.61",
40+
"einops==0.8.0",
41+
"pytorch-lightning==1.4.2",
42+
"torchmetrics==0.5.0",
43+
"ultralytics==8.2.40",
44+
"clip@git+https://github.com/openai/CLIP.git",
45+
"cvias@git+https://github.com/Coargus/Computer-Vision-Inference-and-Server.git@nsvqa-dev",
46+
"vflow@git+https://github.com/Coargus/[email protected]",
4447
]
4548
[tool.black]
4649
line-length = 80
@@ -52,4 +55,3 @@ line-length = 80
5255
[project.optional-dependencies]
5356
dev = ["black", "ruff", "mypy"]
5457
test = ["pytest", "pytest-cov", "pytest-mock"]
55-

0 commit comments

Comments
 (0)