@@ -594,10 +594,8 @@ mod tests {
594594 assert ! ( delete_paths. contains( "build/torch-cuda/inherited.py" ) ) ;
595595 }
596596
597- const METADATA_V3 : & str =
598- r#"{"id": "kernel_id", "version": 3, "python-depends": [], "backend": {"type": "cuda"}}"# ;
599- const METADATA_NO_VERSION : & str =
600- r#"{"id": "kernel_id", "python-depends": [], "backend": {"type": "cuda"}}"# ;
597+ const METADATA_V3 : & str = r#"{"name": "test-kernel", "id": "kernel_id", "version": 3, "license": "Apache-2.0", "python-depends": [], "backend": {"type": "cuda"}}"# ;
598+ const METADATA_V0 : & str = r#"{"name": "test-kernel", "id": "kernel_id", "version": 0, "license": "Apache-2.0", "python-depends": [], "backend": {"type": "cuda"}}"# ;
601599
602600 #[ test]
603601 fn test_detect_branch_from_metadata ( ) {
@@ -612,15 +610,15 @@ mod tests {
612610 }
613611
614612 #[ test]
615- fn test_detect_branch_from_metadata_no_version ( ) {
613+ fn test_detect_branch_from_metadata_v0 ( ) {
616614 let temp_dir = tempfile:: tempdir ( ) . unwrap ( ) ;
617615 let variant = temp_dir. path ( ) . join ( "variant" ) ;
618616 fs:: create_dir_all ( & variant) . unwrap ( ) ;
619- fs:: write ( variant. join ( "metadata.json" ) , METADATA_NO_VERSION ) . unwrap ( ) ;
617+ fs:: write ( variant. join ( "metadata.json" ) , METADATA_V0 ) . unwrap ( ) ;
620618
621619 let variants = vec ! [ variant] ;
622620 let branch = detect_branch_from_metadata ( & variants) . unwrap ( ) ;
623- assert_eq ! ( branch, None ) ;
621+ assert_eq ! ( branch, Some ( "v0" . to_owned ( ) ) ) ;
624622 }
625623
626624 #[ test]
@@ -632,12 +630,12 @@ mod tests {
632630 fs:: create_dir_all ( & v2) . unwrap ( ) ;
633631 fs:: write (
634632 v1. join ( "metadata.json" ) ,
635- r#"{"version": 1, "python-depends": [], "backend": {"type": "cuda"}}"# ,
633+ r#"{"name": "test-kernel", " version": 1, "id": "k1", "license": "Apache-2.0" , "python-depends": [], "backend": {"type": "cuda"}}"# ,
636634 )
637635 . unwrap ( ) ;
638636 fs:: write (
639637 v2. join ( "metadata.json" ) ,
640- r#"{"version": 2, "python-depends": [], "backend": {"type": "cuda"}}"# ,
638+ r#"{"name": "test-kernel", " version": 2, "id": "k2", "license": "Apache-2.0" , "python-depends": [], "backend": {"type": "cuda"}}"# ,
641639 )
642640 . unwrap ( ) ;
643641
@@ -709,6 +707,7 @@ mod tests {
709707 kernel_dir. join ( "build.toml" ) ,
710708 r#"[general]
711709name = "test-kernel"
710+ license = "Apache-2.0"
712711backends = ["cuda"]
713712
714713[general.hub]
@@ -745,6 +744,7 @@ branch = "custom-branch"
745744 kernel_dir. join ( "build.toml" ) ,
746745 r#"[general]
747746name = "test-kernel"
747+ license = "Apache-2.0"
748748backends = ["cuda"]
749749
750750[general.hub]
0 commit comments