Correct the shapes

[benedekrozemberczki]

Summary

• DDI Shapes were mismatched.

• [x] Code passes all tests

• [x] Unit tests provided for these changes

• [x] Documentation and docstrings added for these changes

Changes

• Test modified.

[codecov-commenter]

Codecov Report

Merging #60 (ae5e90b) into main (3c5e640) will increase coverage by 0.22%. The diff coverage is 97.00%.

@@            Coverage Diff             @@
##             main      #60      +/-   ##
==========================================
+ Coverage   97.84%   98.07%   +0.22%     
==========================================
  Files          28       28              
  Lines         744      675      -69     
==========================================
- Hits          728      662      -66     
+ Misses         16       13       -3     

Impacted Files	Coverage Δ
chemicalx/data/datasetloader.py	95.23% <90.69%> (+4.44%)	:arrow_up:
chemicalx/data/batchgenerator.py	98.30% <95.45%> (-0.27%)	:arrow_down:
chemicalx/data/__init__.py	100.00% <100.00%> (ø)
chemicalx/data/contextfeatureset.py	100.00% <100.00%> (ø)
chemicalx/data/drugfeatureset.py	100.00% <100.00%> (ø)
chemicalx/data/drugpairbatch.py	100.00% <100.00%> (ø)
chemicalx/data/labeledtriples.py	97.36% <100.00%> (-0.51%)	:arrow_down:
chemicalx/pipeline.py	88.70% <100.00%> (-1.00%)	:arrow_down:
chemicalx/version.py	100.00% <100.00%> (ø)
tests/unit/test_batching.py	100.00% <100.00%> (ø)
... and 5 more

---

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update 47af8b4...ae5e90b. Read the comment docs.