Skip to content

Commit

Permalink
update unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Hanna Imshenetska authored and Hanna Imshenetska committed Nov 27, 2024
1 parent d220e4a commit c921a9a
Show file tree
Hide file tree
Showing 3 changed files with 426 additions and 2 deletions.
22 changes: 22 additions & 0 deletions src/tests/unit/test_worker/fixtures/metadata_with_reports.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
test_table:
train_settings:
source: ./path/to/test_table.csv
epochs: 100
drop_null: false
reports: all
row_limit: 800
batch_size: 2000

infer_settings:
destination: ./path/to/test_table_infer.csv
size: 200
run_parallel: True
random_seed: 2
reports: all
batch_size: 200

keys:
pk_id:
type: "PK"
columns:
- Id
22 changes: 22 additions & 0 deletions src/tests/unit/test_worker/fixtures/metadata_without_reports.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
test_table:
train_settings:
source: ./path/to/test_table.csv
epochs: 100
drop_null: false
reports: none
row_limit: 800
batch_size: 2000

infer_settings:
destination: ./path/to/test_table_infer.csv
size: 200
run_parallel: True
random_seed: 2
reports: none
batch_size: 200

keys:
pk_id:
type: "PK"
columns:
- Id
Loading

0 comments on commit c921a9a

Please sign in to comment.