Update spark test to avoid warning message

[sfilipi]

Given that the type of data is int, as is, this throws "AttributeError: 'int' object has no attribute 'a' "

Description of proposed changes

Related issue(s)

Fixes # (issue)

Test plan

Checklist

Need help on these? Just ask!

• [ ] I have read the CONTRIBUTING document.
• [ ] I have updated the documentation accordingly.
• [ ] I have added tests to cover my changes.
• [ ] I have run tox -e complex and/or tox -e spark if appropriate.
• [ ] All new and existing tests passed.

[codecov[bot]]

Codecov Report

Merging #1639 (de78a2d) into master (ed77718) will increase coverage by 0.33%. The diff coverage is n/a.

@@            Coverage Diff             @@
##           master    #1639      +/-   ##
==========================================
+ Coverage   97.21%   97.55%   +0.33%     
==========================================
  Files          68       68              
  Lines        2151     2286     +135     
  Branches      345      403      +58     
==========================================
+ Hits         2091     2230     +139     
+ Misses         31       29       -2     
+ Partials       29       27       -2     

Impacted Files	Coverage Δ
snorkel/labeling/model/label_model.py	97.89% <0.00%> (+2.34%)	:arrow_up:

[github-actions[bot]]

This pull request is stale because it has been open 90 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[bhancock8]

Thank you for the PR! Looks good, I'll merge it in.

[paroma]

@sfilipi feel free to merge with a more descriptive PR title when the tests pass!