changed test config file to inherit from default (#49)

* lint

* reconfigured modules in yaml

* added data module

* changed test config file
This commit is contained in:
evonneng authored and GitHub committed 2022-04-29 16:18:30 -07:00
1 parent d79c112248
commit e874e856ec
4 files changed
+24 -7

No files matched your search

+1 -1
View File
@@ -164,7 +164,7 @@ runs/
outputs/
# Data
data/
./data/
# Misc
old/
+1 -1
View File
@@ -20,4 +20,4 @@ machine_config:
num_gpus: 1
num_machines: 1
machine_rank: 0
dist_url: auto
dist_url: auto
+18
View File
@@ -0,0 +1,18 @@
defaults:
- default
dry_run: true
machine_config:
num_gpus: 0
data:
dataset:
data_directory: data/blender/lego_test
validation_image_indices: [0]
dataloader:
num_images_to_sample_from: 1
num_rays_per_batch: 8
graph:
max_num_iterations: 5
+4 -5
View File
@@ -2,8 +2,7 @@
Default test to make sure train runs
"""
import pytest
import omegaconf
from mattport.utils.io import get_absolute_path
from hydra.experimental import compose, initialize
from scripts.run_train import main
@@ -11,6 +10,6 @@ from scripts.run_train import main
@pytest.mark.filterwarnings("ignore::DeprecationWarning")
def test_run_train():
"""test run train script works properly"""
test_config = get_absolute_path("./tests/configs/test_default.yml")
cfg = omegaconf.OmegaConf.load(test_config)
main(cfg)
initialize(config_path="../configs/")
conf = compose("test_dryrun.yaml")
main(conf)